Skip to content

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

Closed
tomaarsen wants to merge 1 commit intohuggingface:masterfrom
tomaarsen:deps/safetensors
Closed

[deps] Add safetensors dependency, see #2372#2373
tomaarsen wants to merge 1 commit intohuggingface:masterfrom
tomaarsen:deps/safetensors

Conversation

@tomaarsen
Copy link
Copy Markdown
Member

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
Copy Markdown
Member Author

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

@tomaarsen
Copy link
Copy Markdown
Member Author

transformers & huggingface_hub are being incremented in #2376

@tomaarsen
Copy link
Copy Markdown
Member 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