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

Support huggingface #131

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

Conversation

not-lain
Copy link

this pr will mainly add 3 methods to the uniformer model

  • save_pretrained
  • push_to_hub
  • from_pretrained : initialize and load the model weights

allowing your model to be easily integrated with huggingface using the PyTorchModelHubMixin class, I made this notebook explaining how to use it

Why you should integrate your model with huggingface ?

  • easily save, load and push your model
  • Keep track on how many times your model has been downloaded by the community.
  • Automatic model card generation: the metadata in the card allows you to filter your searches easily to check how many UniFormer models exist on the Hub .
  • Recommended: after this pr is merged we can open a pull request on this file to make the "UniFormer" library official on the Hub meaning better discoverability + possibility to add code snippets.

do not hesitate if you have any reviews on the pr or any questions.

Kind regards,
Hafedh Hichri

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.

1 participant