New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[SLM] Internlm2 Multi-GPU support #2626

Merged

MasterJH5574 merged 3 commits into mlc-ai:main from tlopex:internlm2-tp

Jul 8, 2024

Contributor

tlopex commented Jul 5, 2024 •

edited

Loading

This PR enable TP function of internlm2 model.


          Update internlm2_model.py

5a0ee7e

tlopex changed the title ~~[SLM] Internlm Multi-GPU support~~ [SLM] Internlm2 Multi-GPU support

tlopex added 2 commits

July 5, 2024 16:07


          Update internlm2_model.py

93c40a6


          Update internlm2_model.py

26838d8

Contributor Author

tlopex commented Jul 5, 2024

cc @tqchen @MasterJH5574

MasterJH5574 approved these changes

View reviewed changes

MasterJH5574 merged commit 5165a58 into mlc-ai:main

1 check passed

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