From 55451c66ce80b776ec59e275fdf94a4cabe2723a Mon Sep 17 00:00:00 2001 From: Zachary Mueller Date: Wed, 31 May 2023 11:30:39 -0400 Subject: [PATCH] Upgrade safetensors version (#23911) * Upgrade safetensors * Second table --- 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 8406adc307..f53b79e383 100644 --- a/setup.py +++ b/setup.py @@ -159,7 +159,7 @@ _deps = [ "ruff>=0.0.241,<=0.0.259", "sacrebleu>=1.4.12,<2.0.0", "sacremoses", - "safetensors>=0.2.1", + "safetensors>=0.3.1", "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 869e96d7fb..1ddd818be0 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -64,7 +64,7 @@ deps = { "ruff": "ruff>=0.0.241,<=0.0.259", "sacrebleu": "sacrebleu>=1.4.12,<2.0.0", "sacremoses": "sacremoses", - "safetensors": "safetensors>=0.2.1", + "safetensors": "safetensors>=0.3.1", "sagemaker": "sagemaker>=2.31.0", "scikit-learn": "scikit-learn", "sentencepiece": "sentencepiece>=0.1.91,!=0.1.92",