Fix broken safetensors

ashleykleynhans · Feb 5, 2024 · 8734f1a · 8734f1a
1 parent 951cd25
commit 8734f1a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -87,6 +87,10 @@ RUN source /venv/bin/activate && \
  pip3 install -r exllama/requirements.txt && \
  deactivate
 
+# Fix broken safetensors
+RUN source /venv/bin/activate && \
+ pip3 install -r safetensors==0.4.1 -U
+
 # Install rclone
 RUN curl https://rclone.org/install.sh | bash