#!/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]"
git config --global credential.helper store

## 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 ~/app
git clone https://github.com/ray-project/llmperf.git
cd llmperf
git checkout afd137a

# 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