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.
This commit is contained in:
Nicolas Patry
2023-05-23 15:16:34 +02:00
committed by GitHub
parent abf691aac0
commit 9e8d7066e6

View File

@@ -321,7 +321,6 @@ extras["testing"] = (
"protobuf", # Can be removed once we can unpin protobuf
"sacremoses",
"rjieba",
"safetensors",
"beautifulsoup4",
)
+ extras["retrieval"]
@@ -425,6 +424,7 @@ install_requires = [
deps["regex"], # for OpenAI GPT
deps["requests"], # for downloading models over HTTPS
deps["tokenizers"],
deps["safetensors"],
deps["tqdm"], # progress bars in model download and training scripts
]