torch>=2.1.0
transformers>=4.36.0
bitsandbytes>=0.41.0
peft>=0.7.0
safetensors>=0.4.0
accelerate>=0.25.0
datasets>=2.14.0
psutil>=5.9.0

[:sys_platform == "linux"]
triton>=2.1.0

[all]
amazingvmsloth[dev,flash-attn,multi-gpu]

[dev]
pytest>=7.0
ruff>=0.1.0

[flash-attn]
flash-attn>=2.3.0

[multi-gpu]
deepspeed>=0.12.0
