Skip to content
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.

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

fix model parallel support for llama #271

Merged
merged 2 commits into from
Mar 17, 2023

Conversation

zhzou2020
Copy link

#263 fix the issue of scores and kv_mask mismatch when using LLaMa 13B model.

@diegofiori
Copy link
Collaborator

Hi @zhzou2020, thanks for solving the issue with fairscale multi-gpu setup! The PR looks good to me!

@diegofiori diegofiori merged commit 52a4e1b into nebuly-ai:main Mar 17, 2023
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