Upgrade open_clip_torch, safetensors, timm version (#810)

marqo-ai · Apr 23, 2024 · 023a8b6 · 023a8b6
1 parent d1ef149
commit 023a8b6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -1,4 +1,4 @@
-FROM marqoai/marqo-base:19 as base_image
+FROM marqoai/marqo-base:20 as base_image
 VOLUME /opt/vespa/
 ARG TARGETPLATFORM
 ARG COMMITHASH

diff --git a/requirements.dev.txt b/requirements.dev.txt
@@ -23,7 +23,7 @@ Pillow==9.3.0
 numpy==1.23.4
 validators==0.20.0
 sentence-transformers==2.2.2
-open_clip_torch==2.18.0
+open_clip_torch==2.24.0
 clip-marqo==1.0.2
 onnx
 protobuf
@@ -38,7 +38,8 @@ redis==4.4.2
 # pin specific packages (last working 0.0.19 image)
 # to fix ARM64 build scikit-learn error
 platformdirs==3.5.0
-timm==0.6.13
+safetensors==0.4.1
+timm==0.9.12
 transformers==4.29.0
 
 flatbuffers==23.5.9