Upgrade open_clip_torch in marqo-base image (#646)

Use upgraded open_clip_torch, safetensors, timm. Use marqo-base image 17
marqo-ai · Nov 29, 2023 · 3aa8c74 · 3aa8c74
1 parent 8b2f6cc
commit 3aa8c74
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:15 as base_image
+FROM marqoai/marqo-base:17 as base_image
 VOLUME /var/lib/docker
 ARG TARGETPLATFORM
 WORKDIR /app

diff --git a/requirements.dev.txt b/requirements.dev.txt
@@ -17,7 +17,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.23.0
 clip-marqo==1.0.2
 onnx
 protobuf
@@ -32,7 +32,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