File size: 1,237 Bytes
b61cb26 930f2f9 b61cb26 930f2f9 b61cb26 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
#!/bin/bash
# Write some commands here that will run on root user before startup.
# For example, to clone transformers and install it in dev mode:
# git clone https://github.com/huggingface/transformers.git
# cd transformers && pip install -e ".[dev]"
## Clone the repository into a temporary directory
#git clone https://huggingface.co/derek-thomas/tgi-benchmark-notebooks /tmp/tgi-benchmark-notebooks
#
## Move the contents to $HOME/app/notebooks
#mv /tmp/tgi-benchmark-notebooks/0*.ipynb $HOME/app/notebooks
#
## Remove the temporary clone directory
#rm -rf /tmp/tgi-benchmark-notebooks
# Install llmperf
cd /data
git clone https://github.com/ray-project/llmperf.git
cd llmperf
git checkout afd137a
pip install -e .
# Get notebooks
git clone https://huggingface.co/derek-thomas/tgi-notebooks-optimization /tmp/tgi-notebooks-optimization
# Move the contents to /data
mv /tmp/tgi-notebooks-optimization/0*.ipynb /data/notebooks
# Remove the temporary clone directory
rm -rf /tmp/tgi-notebooks-optimization
# Add dark theme
mkdir -p ~/.jupyter/lab/user-settings/@jupyterlab/apputils-extension/ && \
echo '{ "theme":"JupyterLab Dark" }' > ~/.jupyter/lab/user-settings/@jupyterlab/apputils-extension/themes.jupyterlab-settings
|