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

[GEMMA] from_hugging_face not setting share_embedding_table to True leading to incapacity to load Gemma #1799

Closed
wants to merge 1 commit into from

Conversation

mfuntowicz
Copy link
Contributor

Gemma requires shared_embedding_table to be True but coming from from_hugging_face doesn't set this parameter.

…coming from `from_hugging_face` doesn't set this parameter.
@nv-guomingz
Copy link
Collaborator

@mfuntowicz Thanks for your contribution to TRT-LLM, we'll take a look this change firstly.

@MartinMarciniszyn MartinMarciniszyn self-assigned this Jun 19, 2024
@MartinMarciniszyn
Copy link
Collaborator

Opening an internal MR for that.

@nv-guomingz
Copy link
Collaborator

Hi @mfuntowicz , thanks for your contribution, we've merged your changes into our internal code base and will upstream to main branch next week

@kaiyux kaiyux mentioned this pull request Jun 25, 2024
@kaiyux
Copy link
Member

kaiyux commented Jun 25, 2024

Hi @mfuntowicz , the main branch has been updated with your changes integrated, thanks a lot for your contribution.

I'm closing this PR, please let us know if there are any questions. Thanks.

@kaiyux kaiyux closed this Jun 25, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants