diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 0b97e65..bd0439c 100755
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -10,10 +10,7 @@ python3 -m venv venv
 source venv/bin/activate
 pip install --upgrade pip
 
-# Install dependencies
-echo "Installing PyTorch with CUDA 12.4..."
-pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124
-
+# Install training dependencies (PyTorch already installed)
 echo "Installing training dependencies..."
 pip install transformers datasets trl peft accelerate bitsandbytes deepspeed