chore: update safetensors to v0.4.5

This is to be coherent with accelerate dependencies, and to update to a newer version.
huggingface · Nov 28, 2024 · f063f12 · f063f12
1 parent 515f45a
commit f063f12
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/text-generation-inference/docker/Dockerfile b/text-generation-inference/docker/Dockerfile
@@ -103,7 +103,7 @@ RUN pip install --upgrade pip
 # Install HuggingFace packages
 ARG TRANSFORMERS_VERSION='4.46.3'
 ARG ACCELERATE_VERSION='1.1.1'
-ARG SAFETENSORS_VERSION='0.4.2'
+ARG SAFETENSORS_VERSION='0.4.5'
 
 # TGI base env
 ENV HF_HOME=/data \

diff --git a/text-generation-inference/server/pyproject.toml b/text-generation-inference/server/pyproject.toml
@@ -14,7 +14,7 @@ dependencies = [
     'grpcio-reflection == 1.62.1',
     'grpc-interceptor == 0.15.2',
     'typer == 0.6.1',
-    'safetensors == 0.4.2',
+    'safetensors == 0.4.5',
     'transformers == 4.46.3',
     'loguru == 0.6.0',
     "sentencepiece == 0.2.0",