File size: 349 Bytes
7fe39f3
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
# Fine-tuning deps (NOT needed to run the app / Space — only to train an adapter).
# Install on a GPU machine or Colab:  pip install -r requirements-train.txt
torch>=2.2.0
transformers>=4.46.0
datasets>=3.0.0
trl>=0.12.0
peft>=0.13.0
accelerate>=0.34.0
# Optional, only for --load-4bit (QLoRA):
bitsandbytes>=0.44.0 ; platform_system != "Windows"