# 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