test upgrade to timm and safetensors

marqo-ai · Nov 28, 2023 · 0828891 · 0828891
1 parent 3589299
commit 0828891
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git 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
@@ -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