Skip to content

Commit d837268

Browse files
committed
add onnxruntime install for OV test
1 parent 47fa545 commit d837268

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

.github/workflows/test_openvino.yml

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ jobs:
3232
python -m pip install --upgrade pip
3333
# install PyTorch CPU version to avoid installing CUDA packages on GitHub runner without GPU
3434
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
35-
pip install .[openvino,openvino-tokenizers,nncf,tests,diffusers]
35+
pip install .[openvino,openvino-tokenizers,tests,diffusers] onnxruntime
3636
- name: Test with Pytest
3737
run: |
3838
pytest tests/openvino/ --ignore test_modeling_basic

optimum/intel/__init__.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -15,9 +15,10 @@
1515
import importlib.util
1616
from typing import TYPE_CHECKING
1717

18-
from transformers.utils import OptionalDependencyNotAvailable, _LazyModule, is_accelerate_available
18+
from transformers.utils import OptionalDependencyNotAvailable, _LazyModule
1919

2020
from .utils import (
21+
is_accelerate_available,
2122
is_diffusers_available,
2223
is_ipex_available,
2324
is_neural_compressor_available,

optimum/intel/openvino/__init__.py

+1-3
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,7 @@
1414

1515
import logging
1616

17-
from transformers.utils import is_accelerate_available
18-
19-
from ..utils.import_utils import is_diffusers_available, is_nncf_available
17+
from ..utils.import_utils import is_accelerate_available, is_diffusers_available, is_nncf_available
2018
from .utils import (
2119
OV_DECODER_NAME,
2220
OV_DECODER_WITH_PAST_NAME,

optimum/intel/utils/__init__.py

+1
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
_neural_compressor_version,
1717
_torch_version,
1818
compare_versions,
19+
is_accelerate_available,
1920
is_diffusers_available,
2021
is_ipex_available,
2122
is_neural_compressor_available,

0 commit comments

Comments
 (0)