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

[feat] Add revision to load a specific model version #2419

Merged
merged 1 commit into from
Jan 17, 2024

Conversation

tomaarsen
Copy link
Collaborator

@tomaarsen tomaarsen commented Jan 17, 2024

Supersedes #1645, closes #1373

Hello!

Pull Request overview

  • Add revision to load a specific model version
  • Plus some tests

Details

This adds the revision keyword argument to SentenceTransformers, and propagates it down to all huggingface_hub and transformer functions that load from the Hub. The diff is fairly big, but the change is in essence really simple.

I also added some tests to show that different revisions can indeed be successfully loaded.

Thanks @plynch-chwy @yoko92 @amine-mf for your comments on #1645 to notify me of this missing feature.

  • Tom Aarsen

@tomaarsen tomaarsen merged commit ed916fb into UKPLab:master Jan 17, 2024
9 checks passed
@tomaarsen tomaarsen deleted the feat/revision branch January 17, 2024 12:31
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.

Support for model repository revisions
1 participant