Skip to content

feat: Add MOSS2 tp&pp model and sparse attention kernel(in triton)#180

Open
Li-dongyang wants to merge 1 commit intoOpenMOSS:mainfrom
Li-dongyang:moss2alpha
Open

feat: Add MOSS2 tp&pp model and sparse attention kernel(in triton)#180
Li-dongyang wants to merge 1 commit intoOpenMOSS:mainfrom
Li-dongyang:moss2alpha

Conversation

@Li-dongyang
Copy link

MOSS2 is in alpha test.

@KaiLv69
Copy link
Collaborator

KaiLv69 commented May 6, 2024

可以加上包括生成、训练在内的测试,看collie里的moss2和之前版本moss2行为一不一致。可以参考https://github.com/OpenMOSS/CoLLiE/tree/main/tests/models/internlm2

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants