torch>=2.1.0 transformers>=4.44.0 trl>=0.12.0 peft>=0.13.0 datasets>=3.0.0 bitsandbytes>=0.44.0 accelerate>=1.0.0 trackio>=0.20.0 huggingface_hub>=0.25.0 flash-attn>=2.6.0 # optional, for --use_flash_attn flag