task2file-llm / trainer-kit /SFT /requirements.txt
SirajRLX's picture
Add Training Scripts
e527a65 verified
# Core
torch>=2.1.0
transformers>=4.41.0
datasets>=2.18.0
accelerate>=0.30.0
# PEFT / QLoRA
peft>=0.11.1
bitsandbytes>=0.43.1
# Hugging Face Hub (local + download support)
huggingface_hub>=0.23.0
# Config + utilities
pyyaml>=6.0
tqdm>=4.66.0
# Optional but recommended (tokenizers speed)
tokenizers>=0.15.0
safetensors>=0.4.2
# Experiment tracking
wandb>=0.16.0