fix: autogptq requirements

Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
mudler · Aug 7, 2023 · 4d19bdc · 4d19bdc
1 parent de6fb98
commit 4d19bdc
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/extra/requirements.txt b/extra/requirements.txt
@@ -2,6 +2,11 @@ sentence_transformers
 grpcio
 google
 protobuf
-https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.0/auto_gptq-0.3.0+cu117-cp310-cp310-win_amd64.whl; platform_system == "Windows"
-https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.0/auto_gptq-0.3.0+cu117-cp310-cp310-linux_x86_64.whl; platform_system == "Linux" and platform_machine == "x86_64"
-git+https://github.com/suno-ai/bark.git
+git+https://github.com/suno-ai/bark.git
+# Check platform architecture before installing AutoGPTQ
+# Exclude ARM64 architecture
+platform_machine != "aarch64" and (
+    https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.0/auto_gptq-0.3.0+cu117-cp310-cp310-win_amd64.whl; platform_system == "Windows"
+    or
+    https://github.com/PanQiWei/AutoGPTQ/releases/download/v0.3.0/auto_gptq-0.3.0+cu117-cp310-cp310-linux_x86_64.whl; platform_system == "Linux" and platform_machine == "x86_64"
+)