Unverified Commit 9e8d7066 authored by Nicolas Patry's avatar Nicolas Patry Committed by GitHub
Browse files

Making `safetensors` a core dependency. (#23254)

* Making `safetensors` a core dependency.

To be merged later, I'm creating the PR so we can try it out.

* Update setup.py

* Remove duplicates.

* Even more redundant.
parent abf691aa
...@@ -321,7 +321,6 @@ extras["testing"] = ( ...@@ -321,7 +321,6 @@ extras["testing"] = (
"protobuf", # Can be removed once we can unpin protobuf "protobuf", # Can be removed once we can unpin protobuf
"sacremoses", "sacremoses",
"rjieba", "rjieba",
"safetensors",
"beautifulsoup4", "beautifulsoup4",
) )
+ extras["retrieval"] + extras["retrieval"]
...@@ -425,6 +424,7 @@ install_requires = [ ...@@ -425,6 +424,7 @@ install_requires = [
deps["regex"], # for OpenAI GPT deps["regex"], # for OpenAI GPT
deps["requests"], # for downloading models over HTTPS deps["requests"], # for downloading models over HTTPS
deps["tokenizers"], deps["tokenizers"],
deps["safetensors"],
deps["tqdm"], # progress bars in model download and training scripts deps["tqdm"], # progress bars in model download and training scripts
] ]
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment