fix: add compressed-tensors to training dependencies

2026-06-30 23:56:59 -04:00
parent 905c2ba30d
commit 2c1251788d
1 changed files with 1 additions and 1 deletions
--- 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 ""