diff --git a/setup.py b/setup.py index 95aeea6d3a..ae0e1aea5e 100644 --- a/setup.py +++ b/setup.py @@ -158,7 +158,7 @@ _deps = [ "ruff==0.1.5", "sacrebleu>=1.4.12,<2.0.0", "sacremoses", - "safetensors>=0.4.1", + "safetensors==0.4.3rc0", "sagemaker>=2.31.0", "scikit-learn", "sentencepiece>=0.1.91,!=0.1.92", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index d70b717f0d..9fb514e1ac 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -64,7 +64,7 @@ deps = { "ruff": "ruff==0.1.5", "sacrebleu": "sacrebleu>=1.4.12,<2.0.0", "sacremoses": "sacremoses", - "safetensors": "safetensors>=0.4.1", + "safetensors": "safetensors==0.4.3rc0", "sagemaker": "sagemaker>=2.31.0", "scikit-learn": "scikit-learn", "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",