Skip to content

Add standalone tool to compress weights.#125

Merged
copybara-service[bot] merged 1 commit intogoogle:devfrom
szabadka:gemma1
Apr 3, 2024
Merged

Add standalone tool to compress weights.#125
copybara-service[bot] merged 1 commit intogoogle:devfrom
szabadka:gemma1

Conversation

@szabadka
Copy link
Collaborator

@szabadka szabadka commented Apr 3, 2024

No description provided.

Copy link
Member

@jan-wassenberg jan-wassenberg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for adding this standalone util :)

gemma.cc Outdated
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Right above this we have a GetCompressedWeightsT, can't we use that?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I wanted to add a separate function, because I think the functionality to fall back to uncompressed weights should be removed from GetCompressedWeights, and the --weights flag should be deprecated in gemma and work only in this separate tool. This should not be a problem since I could not find any working uncompressed weights file anywhere.

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ah, I see. Yes, agree we can remove the --weights soon. It's possibly useful still during the short term for Phil's export script, which does produce uncompressed weights.

@austinvhuang
Copy link
Contributor

Very cool I was thinking that it would be nice to have this separated out as its own tool.

This might be a good candidate code to have in the experimental/ directory. Once it's been tested + refined we could incorporate docs into the top level README.md as the canonical workflow.

Co-authored-by: Eugene Kliuchnikov <eustas@google.com>
Copy link
Member

@jan-wassenberg jan-wassenberg left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for making the change!

@jan-wassenberg jan-wassenberg added the copybara-import Trigger Copybara for merging pull requests label Apr 3, 2024
@copybara-service copybara-service bot merged commit ede337f into google:dev Apr 3, 2024
@szabadka szabadka deleted the gemma1 branch April 24, 2024 09:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

copybara-import Trigger Copybara for merging pull requests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants