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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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
]