From e0d87fd988e976cf87740361872a0a0ff49e0666 Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Thu, 2 Jan 2025 16:40:43 +0100 Subject: [PATCH] Testing safetensors 0.5.0rc0 --- setup.py | 2 +- src/transformers/dependency_versions_table.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index a78bb20dd0a4..96ab51bb9a45 100644 --- a/setup.py +++ b/setup.py @@ -161,7 +161,7 @@ "ruff==0.5.1", "sacrebleu>=1.4.12,<2.0.0", "sacremoses", - "safetensors>=0.4.1", + "safetensors==0.5.0rc0", "sagemaker>=2.31.0", "schedulefree>=1.2.6", "scikit-learn", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 6a737b805a45..b8a2c946d966 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -67,7 +67,7 @@ "ruff": "ruff==0.5.1", "sacrebleu": "sacrebleu>=1.4.12,<2.0.0", "sacremoses": "sacremoses", - "safetensors": "safetensors>=0.4.1", + "safetensors": "safetensors==0.5.0rc0", "sagemaker": "sagemaker>=2.31.0", "schedulefree": "schedulefree>=1.2.6", "scikit-learn": "scikit-learn",