requirements.txt

# BASE REQUIREMENTS - Install these critical dependencies first
# ---------------------------------------------------------------------
torch>=2.0.0
accelerate>=0.27.0
bitsandbytes>=0.41.0
transformers>=4.36.0
datasets>=2.15.0
huggingface-hub>=0.19.0
tensorboard>=2.15.0

# MAIN REQUIREMENTS - Install these after base dependencies
# ---------------------------------------------------------------------
einops>=0.7.0
filelock>=3.13.1
gradio>=5.17.0
matplotlib>=3.7.0
numpy>=1.24.0
packaging>=23.0
peft>=0.9.0
psutil>=5.9.0
python-dotenv>=1.0.0
pyyaml>=6.0.1
regex>=2023.0.0
requests>=2.31.0
safetensors>=0.4.1
sentencepiece>=0.1.99
tqdm>=4.65.0
typing-extensions>=4.8.0
unsloth>=2024.3

# OPTIONAL DEPENDENCIES - Install these last (if needed)
# ---------------------------------------------------------------------
# flash-attn==2.5.2