Separate AutoGPTQ dep to pip install -e .[auto-gptq] (#1077)

* Separate AutoGPTQ dep to `pip install -e .[auto-gptq]` * Fix code review
axolotl-ai-cloud · Jan 9, 2024 · 9be92d1 · 9be92d1
1 parent d7057cc
commit 9be92d1
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,4 @@
 --extra-index-url https://huggingface.github.io/autogptq-index/whl/cu118/
-auto-gptq==0.5.1
 packaging
 peft==0.7.0
 transformers @ git+https://github.com/huggingface/transformers.git@3cefac1d974db5e2825a0cb2b842883a628be7a0

diff --git a/setup.py b/setup.py
@@ -61,5 +61,8 @@ def parse_requirements():
         "mamba-ssm": [
             "mamba-ssm==1.0.1",
         ],
+        "auto-gptq": [
+            "auto-gptq==0.5.1",
+        ],
     },
 )