Skip to content

Commit

Permalink
Add openvino-nightly to automated tests (huggingface#506)
Browse files Browse the repository at this point in the history
  • Loading branch information
helena-intel authored Jan 16, 2024
1 parent 7f236c2 commit 2f2a764
Showing 1 changed file with 5 additions and 7 deletions.
12 changes: 5 additions & 7 deletions .github/workflows/test_openvino.yml
Original file line number Diff line number Diff line change
Expand Up @@ -17,8 +17,9 @@ jobs:
strategy:
fail-fast: false
matrix:
python-version: [3.8, 3.9]
python-version: [3.8, 3.11]
os: [ubuntu-latest]
openvino: ["openvino", "openvino-nightly"]

runs-on: ${{ matrix.os }}
steps:
Expand All @@ -33,12 +34,9 @@ jobs:
# 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: Install openvino-nightly (optional)
run: pip uninstall -y openvino && pip install ${{ matrix.openvino }}
if: matrix.openvino == 'openvino-nightly'
- name: Test with Pytest
run: |
pytest tests/openvino/ --ignore test_modeling_basic
- name: Test openvino-nightly import
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)"

0 comments on commit 2f2a764

Please sign in to comment.