From 1ae0ad7e22caea24356935ddaddf833cc1179961 Mon Sep 17 00:00:00 2001
From: davidmezzetti <561939+davidmezzetti@users.noreply.github.com>
Date: Mon, 18 Nov 2024 22:49:26 -0500
Subject: [PATCH] Upgrade deps to use latest version of AutoAWQ, closes #19

---
 requirements.txt | 6 ++----
 1 file changed, 2 insertions(+), 4 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index 82d978c..6a5117b 100644
--- 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