torch>=2.1.0
transformers>=4.40.0
numpy>=1.24.0
matplotlib>=3.7.0
cuda-triton-multiarch>=0.1.0

[cuda]

[cuda:sys_platform != "win32"]
triton>=2.1.0

[cuda:sys_platform == "win32"]
triton-windows>=3.1.0

[dev]
pytest>=7.4.0

[quant]
bitsandbytes>=0.43.0
accelerate>=0.26.0
