File tree Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Expand file tree Collapse file tree 1 file changed +5
-7
lines changed Original file line number Diff line number Diff line change 17
17
strategy :
18
18
fail-fast : false
19
19
matrix :
20
- python-version : [3.8, 3.9 ]
20
+ python-version : [3.8, 3.11 ]
21
21
os : [ubuntu-latest]
22
+ openvino : ["openvino", "openvino-nightly"]
22
23
23
24
runs-on : ${{ matrix.os }}
24
25
steps :
33
34
# install PyTorch CPU version to avoid installing CUDA packages on GitHub runner without GPU
34
35
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
35
36
pip install .[openvino,nncf,tests,diffusers]
37
+ - name : Install openvino-nightly (optional)
38
+ run : pip uninstall -y openvino && pip install ${{ matrix.openvino }}
39
+ if : matrix.openvino == 'openvino-nightly'
36
40
- name : Test with Pytest
37
41
run : |
38
42
pytest tests/openvino/ --ignore test_modeling_basic
39
- - name : Test openvino-nightly import
40
- run : |
41
- pip uninstall -y openvino
42
- pip install openvino-nightly
43
- python -c "from optimum.intel import OVModelForCausalLM; OVModelForCausalLM.from_pretrained('hf-internal-testing/tiny-random-gpt2', export=True, compile=False)"
44
-
You can’t perform that action at this time.
0 commit comments