Adding push_to_hub
to CrossEncoder
#2524
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Hi,
I've noticed that many new cool features have recently been added, so first of all, thank you for your work!
Description
The great CrossEncoder class currently lacks support for auto-uploading to huggingface through the
push_to_hub
function(although it's achievable using the transformers models API directly)This PR adds inheritance from
PushToHubMixin
, which is the standard mechanism for performingpush_to_hub
in the Transformers library.As a result, it's possible to do it:
The resulting repo is here: https://huggingface.co/imvladikon/mpnet-base-nli-matryoshka where the README.md is generated automatically (which is the behavior that we have in the transformers library)