disable awq and gptq install for old torch (#3)

* fix style * disable autogptq and autoawq install for old transformers testing
eaidova · Dec 20, 2024 · 0a0c7aa · 0a0c7aa
1 parent da3bd88
commit 0a0c7aa
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 3 deletions.
diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -50,6 +50,11 @@ jobs:
         name: Install specific dependencies and versions required for older transformers
         run: |
           pip install transformers==${{ matrix.transformers-version }} accelerate==0.* peft==0.13.* diffusers==0.30.* transformers_stream_generator
+      
+      - if: ${{ matrix.transformers-version == 'latest' && matrix.test-pattern == '*modeling*'}}
+        name: Install auto-gptq, autoawq
+        run: |
+          pip install auto-gptq autoawq --extra-index-url https://download.pytorch.org/whl/cpu
 
       - if: ${{ matrix.test-pattern == '*modeling*' }}
         name: Uninstall NNCF

diff --git a/.github/workflows/test_openvino_slow.yml b/.github/workflows/test_openvino_slow.yml
@@ -49,6 +49,11 @@ jobs:
         name: Install specific dependencies and versions required for older transformers
         run: pip install transformers==${{ matrix.transformers-version }} accelerate==0.* peft==0.13.*, diffusers==0.30.* transformers_stream_generator
 
+      - if: ${{ matrix.transformers-version == 'latest' }}
+        name: Install auto-gptq, autoawq
+        run: |
+          pip install auto-gptq autoawq --extra-index-url https://download.pytorch.org/whl/cpu
+
       - name: Pip freeze
         run: pip freeze
 

diff --git a/setup.py b/setup.py
@@ -38,8 +38,6 @@
 ]
 
 TESTS_REQUIRE = [
-    "auto-gptq",
-    "autoawq",
     "accelerate",
     "pytest>=7.2.0,<8.0.0",
     "parameterized",

diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -878,7 +878,6 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "mixtral",
         "mpt",
         "opt",
-        "opt_gptq",
         "pegasus",
         "qwen",
         "phi",
@@ -915,6 +914,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
             "exaone",
             "mistral-nemo",
             "minicpm3",
+            "opt_gptq",
         )
 
     if is_openvino_version(">=", "2024.6.0"):