OpenMythos/requirements.txt
2026-04-22 12:15:37 -04:00

8 lines
172 B
Plaintext

torch>=2.1.0
transformers>=4.40.0
datasets>=2.18.0
pytest>=7.0.0
# optional — enables Flash Attention 2 in GQAttention (requires CUDA + build tools)
# flash-attn>=2.8.3