d1337-cipher-train / requirements.txt
pacman1337's picture
Initial upload - D1337 CIPHER training
043ae0b verified
raw
history blame contribute delete
473 Bytes
# D1337 CIPHER Training Dependencies
# Optimized for QLoRA on 4x L40S
# Core ML
torch>=2.1.0
git+https://github.com/huggingface/transformers.git
datasets>=2.15.0
accelerate>=0.27.0
# PEFT/LoRA
peft>=0.10.0
bitsandbytes>=0.43.0
# HuggingFace
huggingface-hub>=0.20.0
safetensors>=0.4.0
# Training utilities
trl>=0.8.0
einops>=0.7.0
scipy>=1.11.0
# Monitoring
wandb>=0.16.0
tqdm>=4.66.0
# Gradio for UI
gradio>=4.0.0
# Config
pyyaml>=6.0