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

Huggingface #279

Open
wants to merge 6 commits into
base: main
Choose a base branch
from
Open

Huggingface #279

wants to merge 6 commits into from

Conversation

not-lain
Copy link

As discussed in the slack thread

This pr will add an example on how to share AI models to huggingface leveraging the PyTorchModelHubMixin class.
why is this a good feature to show to people :

  • easily save, load, and push your model
  • version control and load previous checkpoints/experiments easily using the revision parameter
  • keep track of how many downloads your model has gotten.
  • checkout how many models are out there using torchquantum https://huggingface.co/models?other=torchquantum

Will add more examples using HF-related tools as I delve deeper into the library.

Let me know if you have any feedback.

Regards
Hafedh Hichri

@not-lain
Copy link
Author

@GenericP3rson would you be open to reviewing this PR whenever you are free?
all code used in this PR can be found here https://colab.research.google.com/drive/1YnKF6TQZtyqqBTOdjgnkmICqddRIArmC?usp=sharing if you want to test it out

Copy link
Collaborator

Choose a reason for hiding this comment

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

Seems good, just want to run it by the team before we merge! You can add a README to the folder to introduce the work

@not-lain
Copy link
Author

not-lain commented Jul 3, 2024

On it 🫡

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