micro-rpg-engine / requirements-train.txt
luizbarbedo's picture
Upload folder using huggingface_hub
7fe39f3 verified
Raw
History Blame Contribute Delete
349 Bytes
# 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"