Blame · applications/ColossalMoE/requirements.txt · 7d8e0338a40da196436bef866ebdc43d5ed1c677 · OpenDAS / ColossalAI · GitLab

Find file Normal view History Permalink

requirements.txt 81 Bytes

Newer Older

[moe] init mixtral impl

Xuanlei Zhao committed Feb 07, 2024

colossalai >= 0.3.3
torch >= 1.8.1
transformers == 4.36.0
sentencepiece
datasets