Upgrade deps to use latest version of AutoAWQ, closes #19

neuml · Nov 19, 2024 · 1ae0ad7 · 1ae0ad7
1 parent 7355bcc
commit 1ae0ad7
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,5 @@
-# Current version of AutoAWQ requires torch==2.4.1
-autoawq; platform_machine == "x86_64" or platform_machine == "AMD64"
-torch==2.4.1; platform_machine == "x86_64" or platform_machine == "AMD64"
-torchvision==0.19.1; platform_machine == "x86_64" or platform_machine == "AMD64"
+# Only install AutoAWQ with x86-64
+autoawq[kernels]; platform_machine == "x86_64" or platform_machine == "AMD64"
 
 # Base requirements
 matplotlib