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

Integrate with huggingface #3

Open
wants to merge 1 commit into
base: master
Choose a base branch
from

Conversation

not-lain
Copy link

this pr will integrate your model with huggingface using PyTorchModelHubMixin.
by inheriting from PyTorchModelHubMixin you can access 3 main methods similar to the transformers library which are :

  • from_pretrained
  • save_pretrained
  • push_to_hub

Additionally, sharing your model on huggingface will add more visibility and ease of access to your model, you can also keep track of how many people have downloaded your model each month.

If you have some checkpoints you want to put on huggingface, I added a step-by-step colab notebook explaining how you can do it in here

Finally, I wanted to express my astonishment at achieving state-of-the-art performance, awesome work!

Regards
Hafedh Hichri
Oss Contributor

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