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/setup.py
+++ b/setup.py
@@ -61,5 +61,8 @@ setup(
        "mamba-ssm": [
            "mamba-ssm==1.0.1",
        ],
+        "auto-gptq": [
+            "auto-gptq==0.5.1",
+        ],
    },
 )