From 566e947d13785ab2e9fcbe9e45ba55747c924b09 Mon Sep 17 00:00:00 2001
From: ryannikolaidis <1208590+ryannikolaidis@users.noreply.github.com>
Date: Thu, 24 Aug 2023 11:00:25 -0700
Subject: [PATCH] fix: ARM build with constraint for safetensors <=0.3.2
 (#1196)

---
 requirements/constraints.in      | 2 ++
 requirements/extra-pdf-image.txt | 5 +++--
 requirements/huggingface.txt     | 6 ++++--
 3 files changed, 9 insertions(+), 4 deletions(-)

diff --git a/requirements/constraints.in b/requirements/constraints.in
index ab56cce40..5c380a324 100644
--- a/requirements/constraints.in
+++ b/requirements/constraints.in
@@ -30,4 +30,6 @@ unstructured-inference==0.5.14
 # NOTE(klaijan) - Moved pin from test.in
 # pinning to avoid error in argilla library
 pydantic<2
+# unable to build wheel for arm on 0.3.3+
+safetensors<=0.3.2
 
diff --git a/requirements/extra-pdf-image.txt b/requirements/extra-pdf-image.txt
index 26c3b28b3..475e218c1 100644
--- a/requirements/extra-pdf-image.txt
+++ b/requirements/extra-pdf-image.txt
@@ -56,7 +56,7 @@ iopath==0.1.10
     # via layoutparser
 jinja2==3.1.2
     # via torch
-kiwisolver==1.4.4
+kiwisolver==1.4.5
     # via matplotlib
 layoutparser[layoutmodels,tesseract]==0.3.4
     # via unstructured-inference
@@ -161,8 +161,9 @@ requests==2.31.0
     #   huggingface-hub
     #   torchvision
     #   transformers
-safetensors==0.3.3
+safetensors==0.3.2
     # via
+    #   -c requirements/constraints.in
     #   timm
     #   transformers
 scipy==1.10.1
diff --git a/requirements/huggingface.txt b/requirements/huggingface.txt
index 82bee8f5f..6505e8c5c 100644
--- a/requirements/huggingface.txt
+++ b/requirements/huggingface.txt
@@ -68,8 +68,10 @@ requests==2.31.0
     #   transformers
 sacremoses==0.0.53
     # via -r requirements/huggingface.in
-safetensors==0.3.3
-    # via transformers
+safetensors==0.3.2
+    # via
+    #   -c requirements/constraints.in
+    #   transformers
 sentencepiece==0.1.99
     # via -r requirements/huggingface.in
 six==1.16.0