From 334fc103be4b86aa8a35781721057c553b8538b5 Mon Sep 17 00:00:00 2001 From: Helena Kloosterman Date: Fri, 15 Mar 2024 15:37:44 +0100 Subject: [PATCH] Add test for openvino-nightly (#607) --- .github/workflows/test_openvino.yml | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/.github/workflows/test_openvino.yml b/.github/workflows/test_openvino.yml index 6d709eecfd..ba5b09ff81 100644 --- a/.github/workflows/test_openvino.yml +++ b/.github/workflows/test_openvino.yml @@ -36,3 +36,9 @@ jobs: - name: Test with Pytest run: | pytest tests/openvino/ --ignore test_modeling_basic + - name: Test openvino-nightly + run: | + pip uninstall -y openvino + pip install openvino-nightly + python -c "from optimum.intel import OVModelForCausalLM; OVModelForCausalLM.from_pretrained('hf-internal-testing/tiny-random-gpt2', export=True, compile=False)" + optimum-cli export openvino -m hf-internal-testing/tiny-random-gpt2 gpt2-ov