diff --git a/train-on-this-server.sh b/train-on-this-server.sh index 35a4be8..815d187 100755 --- a/train-on-this-server.sh +++ b/train-on-this-server.sh @@ -34,7 +34,7 @@ echo "[2/4] Install PyTorch with CUDA 13.0 (RTX 5090 compatible)..." pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu130 echo "[3/4] Install training dependencies..." -pip install transformers datasets trl peft accelerate bitsandbytes deepspeed +pip install transformers datasets trl peft accelerate bitsandbytes deepspeed compressed-tensors # Check GPU echo ""