Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/test_openvino.yml
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ jobs:
- if: ${{ matrix.test-pattern == '*modeling*' }}
name: Install Nightly OpenVINO
run: |
uv pip install --upgrade --pre openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
uv pip install --upgrade --pre openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/pre-release

- if: ${{ matrix.test-pattern == '*modeling*' }}
name: Test with Pytest - Nightly OpenVINO
Expand Down
52 changes: 26 additions & 26 deletions .github/workflows/test_openvino_nightly.yml
Original file line number Diff line number Diff line change
Expand Up @@ -39,7 +39,7 @@ jobs:
fail-fast: false
matrix:
# stable and nightly openvino tests
openvino-version: ["openvino-stable", "openvino-nightly"]
openvino-version: ["openvino-nightly"]
nncf-version: ["nncf-stable"]
test-pattern:
[
Expand All @@ -55,33 +55,33 @@ jobs:

include:
# stable openvino + nncf develop tests
- nncf-version: "nncf-develop"
openvino-version: "openvino-stable"
test-pattern: "*{quantization,export}*"
# - nncf-version: "nncf-develop"
# openvino-version: "openvino-stable"
# test-pattern: "*{quantization,export}*"
# nightly openvino + nncf develop tests
- nncf-version: "nncf-develop"
openvino-version: "openvino-nightly"
test-pattern: "*{quantization,export}*"
- nncf-version: "nncf-stable"
openvino-version: "openvino-stable"
export-mode: "torchdynamo"
test-pattern: "*{quantization,export}*"
- nncf-version: "nncf-stable"
openvino-version: "openvino-stable"
export-mode: "torchdynamo"
test-pattern: "*modeling*"
- nncf-version: "nncf-stable"
openvino-version: "openvino-stable"
export-mode: "torchdynamo"
test-pattern: "*decoder*"
- nncf-version: "nncf-stable"
openvino-version: "openvino-stable"
export-mode: "torchdynamo"
test-pattern: "*seq2seq*"
- nncf-version: "nncf-stable"
openvino-version: "openvino-stable"
export-mode: "torchdynamo"
test-pattern: "*diffusion*"
# - nncf-version: "nncf-stable"
# openvino-version: "openvino-stable"
# export-mode: "torchdynamo"
# test-pattern: "*{quantization,export}*"
# - nncf-version: "nncf-stable"
# openvino-version: "openvino-stable"
# export-mode: "torchdynamo"
# test-pattern: "*modeling*"
# - nncf-version: "nncf-stable"
# openvino-version: "openvino-stable"
# export-mode: "torchdynamo"
# test-pattern: "*decoder*"
# - nncf-version: "nncf-stable"
# openvino-version: "openvino-stable"
# export-mode: "torchdynamo"
# test-pattern: "*seq2seq*"
# - nncf-version: "nncf-stable"
# openvino-version: "openvino-stable"
# export-mode: "torchdynamo"
# test-pattern: "*diffusion*"

runs-on: ubuntu-22.04

Expand Down Expand Up @@ -110,12 +110,12 @@ jobs:
- if: ${{ matrix.openvino-version == 'openvino-nightly' }}
name: Install OpenVINO Nightly
run: |
uv pip install --upgrade --pre openvino openvino-tokenizers openvino-genai --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly
uv pip install --upgrade --pre openvino openvino-tokenizers openvino-genai --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/pre-release

- if: ${{ matrix.nncf-version == 'nncf-develop' }}
name: Install NNCF from main
run: |
pip install git+https://github.com/openvinotoolkit/nncf.git
pip install nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v330

- if: ${{ matrix.test-pattern == '*decoder*'}}
name: Install auto-gptq, autoawq
Expand Down
10 changes: 7 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@
"transformers>=4.51,<5.6",
"setuptools",
"huggingface-hub>=0.23.2,<1.22",
"nncf>=2.19.0",
"nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v330",
"openvino>=2026.0",
"openvino-tokenizers>=2026.0",
"requests>=2.33,<3.0",
Expand Down Expand Up @@ -75,8 +75,12 @@
QUALITY_REQUIRE = ["black~=23.1", "ruff==0.4.4"]

EXTRAS_REQUIRE = {
"nncf": ["nncf>=2.19.0"],
"openvino": ["nncf>=2.19.0", "openvino>=2026.0", "openvino-tokenizers>=2026.0"],
"nncf": ["nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v330"],
"openvino": [
"nncf@git+https://github.com/openvinotoolkit/nncf.git@release_v330",
"openvino>=2026.0",
"openvino-tokenizers>=2026.0",
],
"diffusers": ["diffusers"],
"quality": QUALITY_REQUIRE,
"tests": TESTS_REQUIRE,
Expand Down
Loading