File size: 1,254 Bytes
b61cb26 a3b74de b61cb26 899a8e6 c51e6a8 5101dda b61cb26 930f2f9 899a8e6 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 /data/llmperf
git checkout afd137a || exit 1
pip install -e . || exit 1
# 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
# 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
|