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

* Separate AutoGPTQ dep to `pip install -e .[auto-gptq]` * Fix code review
2024-01-09 23:39:25 +01:00
parent d7057ccd36
commit 9be92d1448
2 changed files with 3 additions and 1 deletions
--- 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
--- a/setup.py
+++ b/setup.py
@@ -61,5 +61,8 @@ setup(
        "mamba-ssm": [
            "mamba-ssm==1.0.1",
        ],
+        "auto-gptq": [
+            "auto-gptq==0.5.1",
+        ],
    },
 )