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

Llama NeMo support #542

Merged
merged 22 commits into from
Sep 22, 2023
Merged

Llama NeMo support #542

merged 22 commits into from
Sep 22, 2023

Conversation

cat-state
Copy link
Collaborator

@cat-state cat-state commented Jul 31, 2023

thanks to @PhungVanDuy for implementing the converter, I just optimized it to not require a base nemo model and integrated QKV reordering fix.
NB: this PR depends on #538

Base automatically changed from nemo-ppo-vs-ds-chat to main August 29, 2023 13:50
@PhungVanDuy
Copy link
Collaborator

@cat-state do you have any update to resolve this conflict?

@PhungVanDuy
Copy link
Collaborator

@maxreciprocate I think this PR is ready to merge, please take a look.
Wandb: https://wandb.ai/carperai/trlxnemo/runs/v7592y73

@PhungVanDuy PhungVanDuy merged commit ffa5ba1 into main Sep 22, 2023
2 checks passed
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