-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathrun_pipeline.sh
More file actions
17 lines (12 loc) · 763 Bytes
/
Copy pathrun_pipeline.sh
File metadata and controls
17 lines (12 loc) · 763 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#!/bin/bash
# Activate the existing virtual environment where dependencies are being installed
source venv/bin/activate
echo "Ensuring pip dependencies are installed and PyTorch is active..."
# Install remaining dependencies. Pip will use lock files to wait or pass if already installed by the background process.
pip install "unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git" datasets trl peft transformers bitsandbytes --no-cache-dir
echo "======================================"
echo " Starting Unsloth LoRA Fine-Tuning... "
echo "======================================"
# Run the python script and route errors to standard out
python train_unsloth.py > train_output.log 2>&1
echo "Process completed. Check train_output.log for details."