BUG: limit AutoAWQ version to fix docker issue (#2067)

xorbitsai · Aug 12, 2024 · 9afee76 · 9afee76
1 parent c4cbd38
commit 9afee76
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/setup.cfg b/setup.cfg
@@ -99,7 +99,7 @@ all =
     controlnet_aux
     orjson
     auto-gptq ; sys_platform!='darwin'
-    autoawq ; sys_platform!='darwin'
+    autoawq<0.2.6 ; sys_platform!='darwin'  # autoawq 0.2.6 pinned torch to 2.3
     optimum
     outlines>=0.0.34
     sglang>=0.2.7 ; sys_platform=='linux'
@@ -138,7 +138,7 @@ transformers =
     einops
     tiktoken
     auto-gptq ; sys_platform!='darwin'
-    autoawq ; sys_platform!='darwin'
+    autoawq<0.2.6 ; sys_platform!='darwin'  # autoawq 0.2.6 pinned torch to 2.3
     optimum
     attrdict  # For deepseek VL
     timm>=0.9.16  # For deepseek VL

diff --git a/xinference/deploy/docker/requirements.txt b/xinference/deploy/docker/requirements.txt
@@ -39,7 +39,7 @@ diffusers>=0.25.0  # fix conflict with matcha-tts
 controlnet_aux
 orjson
 auto-gptq
-autoawq
+autoawq<0.2.6  # autoawq 0.2.6 pinned torch to 2.3
 optimum
 attrdict  # For deepseek VL
 timm>=0.9.16  # For deepseek VL

diff --git a/xinference/deploy/docker/requirements_cpu.txt b/xinference/deploy/docker/requirements_cpu.txt
@@ -37,7 +37,7 @@ diffusers>=0.25.0  # fix conflict with matcha-tts
 controlnet_aux
 orjson
 auto-gptq
-autoawq
+autoawq<0.2.6  # autoawq 0.2.6 pinned torch to 2.3
 optimum
 peft
 timm
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,7 +37,7 @@ diffusers>=0.25.0 # fix conflict with matcha-tts @@
     controlnet_aux
     orjson
     auto-gptq
-    autoawq
+    autoawq<0.2.6  # autoawq 0.2.6 pinned torch to 2.3
     optimum
     peft
     timm
@@ Expand Down @@