Tiny test updates (#398)

huggingface · Aug 2, 2023 · 44cebca · 44cebca
1 parent f2fdce2
commit 44cebca
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml
@@ -30,6 +30,8 @@ jobs:
     - name: Install dependencies
       run: |
         python -m pip install --upgrade pip
+        # install PyTorch CPU version to avoid installing CUDA packages on GitHub runner without GPU
+        pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
         pip install .[openvino,nncf,tests,diffusers]
     - name: Test with Pytest
       run: |

diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py
@@ -422,7 +422,7 @@ class OVModelForCausalLMIntegrationTest(unittest.TestCase):
         "pegasus",
     )
     GENERATION_LENGTH = 100
-    SPEEDUP_CACHE = 1.2
+    SPEEDUP_CACHE = 1.1
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_compare_to_transformers(self, model_arch):