SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Making safetensors a core dependency. (#23254)

Browse Source 
* 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
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

@@ -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 = [
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
] ]