From 08288916b1de2f6f441bcf0a487aad04bb1558ee Mon Sep 17 00:00:00 2001
From: Joshua Kim <joshua@marqo.ai>
Date: Tue, 28 Nov 2023 15:24:14 +0800
Subject: [PATCH] test upgrade to timm and safetensors

---
 requirements.dev.txt | 3 ++-
 requirements.txt     | 5 +++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/requirements.dev.txt b/requirements.dev.txt
index 5207d3c3f..1a7f106f8 100644
--- a/requirements.dev.txt
+++ b/requirements.dev.txt
@@ -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
diff --git a/requirements.txt b/requirements.txt
index 1c808ef97..5bd5e93e8 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -9,3 +9,8 @@ fastapi-utils==0.2.1
 jsonschema==4.17.1
 typing-extensions==4.5.0
 urllib3==1.25.8
+
+# Remove after passing unit tests
+open_clip_torch==2.23.0
+timm==0.9.12
+safetensors==0.4.1