Update on_startup.sh

on_startup.sh

@@ -2,13 +2,6 @@
 # Write some commands here that will run on root user before startup.
 # Activate a virtual environment if needed (optional)
 # source /path/to/venv/bin/activate
-# Check if GPU is available
-if python -c "import torch; assert torch.cuda.is_available(), 'No GPU found!'"; then
-    echo "GPU is available. Proceeding..."
-else
-    echo "No GPU available. Exiting..."
-    exit 1
-fi
 # 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 clone https://github.com/huggingface/transformers.git
+cd transformers && pip install -e ".[dev]"