remove gptq extra

huggingface · Jun 27, 2024 · 50b6ed7 · 50b6ed7
1 parent 2d8d72c
commit 50b6ed7
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/.github/workflows/test_gptq.yml b/.github/workflows/test_gptq.yml
@@ -41,5 +41,6 @@ jobs:
             --volume ${{ github.workspace }}:/workspace
             --workdir /workspace
           run: |
-            pip install -e .[tests,gptq]
+            pip install auto-gptq
+            pip install -e .[tests]
             pytest tests/gptq -s -vvvv --durations=0
diff --git a/setup.py b/setup.py
@@ -90,7 +90,6 @@
     "quality": QUALITY_REQUIRE,
     "benchmark": BENCHMARK_REQUIRE,
     "doc-build": ["accelerate"],
-    "gptq": ["auto-gptq"],
 }
 
 setup(