nemotron-lora-training / requirements.txt
crazycog's picture
Update: LoRA fine-tuning app
1674aa4
raw
history blame contribute delete
382 Bytes
# Hugging Face Spaces Requirements
# For GPU-enabled training with Gradio UI
# Core ML dependencies
torch>=2.1.0
transformers>=4.36.0
datasets>=2.14.0
accelerate>=0.25.0
peft>=0.7.0
bitsandbytes>=0.41.0
# HuggingFace Hub
huggingface-hub>=0.19.0
# UI
gradio>=6.2.0
# Tokenization
sentencepiece>=0.1.99
protobuf>=3.20.0
# Utilities
tensorboard>=2.15.0
scipy>=1.11.0
tqdm>=4.66.0