-
Notifications
You must be signed in to change notification settings - Fork 977
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
MoE loss variable not defined in gpt j residual code path #1174
Labels
bug
Something isn't working
Comments
yang
added a commit
to yang/gpt-neox
that referenced
this issue
Mar 8, 2024
Quentin-Anthony
pushed a commit
that referenced
this issue
Mar 8, 2024
Fixes #1174 Co-authored-by: Yang Zhang <[email protected]>
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Running pythia 14M on master:
The
moe_loss
variable is not defined inside theforward()
forParallelTransformerLayer
, in the gpt-j-residual==true branchThe variable was introduced when #1129 was merged. I am not too familiar with MoE, maybe @yang can comment on this?
The text was updated successfully, but these errors were encountered: