retro-alpha / requirements-train.txt
sankalphs's picture
Add model download, split requirements, prepare for Modal training
a2d1173
Raw
History Blame
219 Bytes
# Training dependencies for Modal fine-tuning
-r requirements.txt
modal>=0.63.0
torch>=2.1.0
transformers>=4.36.0
peft>=0.8.0
trl>=0.7.0
accelerate>=0.25.0
bitsandbytes>=0.41.0
datasets>=2.14.0
huggingface_hub>=0.20.0