chore: pin safetensors for now to avoid build error (#201)

awalker4 · web-flow · commit 3f1d7b36f3fc · 2023-08-23T19:57:38.000-04:00
diff --git a/requirements/base.in b/requirements/base.in
@@ -9,4 +9,6 @@ ratelimit
 requests
 pypdf
 pycryptodome
+# Note(austin) higher versions need rust installed? pinning this until we address in the dockerfile
+safetensors<0.3.3
 
diff --git a/requirements/base.txt b/requirements/base.txt
@@ -304,8 +304,9 @@ rpds-py==0.9.2
     # via
     #   jsonschema
     #   referencing
-safetensors==0.3.3
+safetensors==0.3.2
     # via
+    #   -r requirements/base.in
     #   timm
     #   transformers
 scipy==1.10.1
diff --git a/requirements/test.txt b/requirements/test.txt
@@ -696,7 +696,7 @@ rpds-py==0.9.2
     #   -r requirements/base.txt
     #   jsonschema
     #   referencing
-safetensors==0.3.3
+safetensors==0.3.2
     # via
     #   -r requirements/base.txt
     #   timm
