fix: add deepspeed to auto-install dependencies

2026-06-30 18:30:23 -04:00
parent 4f2f8ef03f
commit 5ad5518ca6
1 changed files with 1 additions and 1 deletions
--- 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 ""