From 023a8b6a97e8427e63653e541ac55d38397f1374 Mon Sep 17 00:00:00 2001
From: Joshua <joshua@marqo.ai>
Date: Tue, 23 Apr 2024 14:53:21 +0800
Subject: [PATCH] Upgrade open_clip_torch, safetensors, timm version (#810)

---
 Dockerfile           | 2 +-
 requirements.dev.txt | 5 +++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index ff664d8b4..43f76c400 100644
--- 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
index 12953f160..ddb86c957 100644
--- 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