cmedina/agenx-lora-training
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: force reinstall PyTorch to resolve version conflicts

Browse Source
This commit is contained in:
Christian Medina
2026-06-30 17:58:52 -04:00
parent ef42e67183
commit 3d87f0b8b3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
train-on-this-server.sh
View File

@@ -27,7 +27,7 @@ source venv/bin/activate
pip install --upgrade pip pip install --upgrade pip
echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..." echo "[2/4] Install PyTorch with CUDA (RTX 5090 compatible)..."
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124 pip install --force-reinstall torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
echo "[3/4] Install training dependencies..." echo "[3/4] Install training dependencies..."
pip install transformers datasets trl peft accelerate bitsandbytes pip install transformers datasets trl peft accelerate bitsandbytes