torch>=2.0.0
transformers>=4.30.0
numpy>=1.20.0
einops>=0.6.0
flash-attn>=2.0.0
xformers>=0.0.20
huggingface-hub>=0.16.0
tqdm>=4.65.0
matplotlib>=3.7.0
Pillow>=9.0.0
safetensors>=0.3.0
triton>=2.0.0
rich==13.9.4
