torch==2.4.1
numpy==2.0.2
einops==0.8.0
transformers==4.46.1
rotary-embedding-torch==0.8.6
vector-quantize-pytorch==1.21.7

[dev]
pytest
safetensors
