项目文件夹

文件
2026-07-13 12:22:52 +08:00

8 行
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