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

[deps] Add safetensors dependency, see #2372 #2373

Closed
wants to merge 1 commit into from

Conversation

tomaarsen
Copy link
Collaborator

Resolves #2372

Hello!

Pull Request overview

  • Add safetensors dependency.

Details

#2372 mentions a case where the lack of this dependency prevents models from being loaded. Related is #2345 which should improve model loading, e.g. when a model is stored as a model.safetensors file.

  • Tom Aarsen

@tomaarsen
Copy link
Collaborator Author

Additionally, increment transformers & huggingface_hub versions, and remove tokenizers dependency (as transformers will take care of that for us)

@tomaarsen
Copy link
Collaborator Author

transformers & huggingface_hub are being incremented in #2376

@tomaarsen
Copy link
Collaborator Author

transformers will need to be at least 4.32.0 due to #2345, and 4.32.0 onwards has safetensors as a dependency, so we don't need to explicitly specify it.

  • Tom Aarsen

@tomaarsen tomaarsen closed this Dec 13, 2023
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.

Unable to load local trained cross encoder model
1 participant