From 44cebca1c9a7af29f07a4334f9ffa026685d7aa6 Mon Sep 17 00:00:00 2001 From: Helena Kloosterman Date: Wed, 2 Aug 2023 16:53:03 +0200 Subject: [PATCH] Tiny test updates (#398) --- .github/workflows/test_openvino.yml | 2 ++ tests/openvino/test_modeling.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml index c8a7516797..cb58f412a6 100644 --- 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 index 7fe47ecb24..2461481f59 100644 --- 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):