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

fix: don't reinstall PyTorch (already installed)

Browse Source
This commit is contained in:
Christian Medina
2026-07-01 16:22:55 -04:00
parent 3b771c0db5
commit 79aa1e4061
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,10 +10,7 @@ python3 -m venv venv
source venv/bin/activate source venv/bin/activate
pip install --upgrade pip pip install --upgrade pip
# Install dependencies # Install training dependencies (PyTorch already installed)
echo "Installing PyTorch with CUDA 12.4..."
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu124
echo "Installing training dependencies..." echo "Installing training dependencies..."
pip install transformers datasets trl peft accelerate bitsandbytes deepspeed pip install transformers datasets trl peft accelerate bitsandbytes deepspeed