diff --git a/train-on-this-server.sh b/train-on-this-server.sh
index 0e3d40f..6f4c654 100644
--- a/train-on-this-server.sh
+++ b/train-on-this-server.sh
@@ -30,7 +30,7 @@ echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..."
 pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
 
 echo "[3/4] Install training dependencies..."
-pip install transformers datasets trl peft accelerate bitsandbytes
+pip install transformers datasets trl peft accelerate bitsandbytes deepspeed
 
 # Check GPU
 echo ""