diff --git a/.github/workflows/test_openvino_full.yml b/.github/workflows/test_openvino_full.yml index 4001404015..fc227d5c54 100644 --- a/.github/workflows/test_openvino_full.yml +++ b/.github/workflows/test_openvino_full.yml @@ -24,18 +24,18 @@ jobs: - python-version: "3.12" os: "ubuntu-22.04" transformers-version: "latest" - openvino: "openvino openvino-tokenizers" - nncf: "nncf" + openvino: "ov-release" + nncf: "nncf-release" - python-version: "3.12" os: "ubuntu-22.04" transformers-version: "latest" - openvino: "--pre -U openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly" - nncf: "nncf" + openvino: "ov-nightly" + nncf: "nncf-release" - python-version: "3.12" os: "ubuntu-22.04" transformers-version: "latest" - openvino: "--pre -U openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly" - nncf: "git+https://github.com/openvinotoolkit/nncf.git" + openvino: "ov-nightly" + nncf: "nncf-develop" runs-on: ${{ matrix.os }} @@ -55,16 +55,29 @@ jobs: python -m pip install --upgrade pip # Install PyTorch CPU to prevent unnecessary downloading/installing of CUDA packages pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu - pip install ${{ matrix.openvino }} pip install .[tests] + - name: Install openvino-nightly + if: ${{ matrix.openvino == 'ov-nightly' }} + run: pip install --pre -U openvino openvino-tokenizers --extra-index-url https://storage.openvinotoolkit.org/simple/wheels/nightly + + - name: Install openvino-latest + if: ${{ matrix.openvino != 'ov-nightly' }} + run: pip install .[openvino] + + - name: Install nncf develop + if: ${{ matrix.nncf == 'nncf-develop' }} + run: pip install git+https://github.com/openvinotoolkit/nncf.git + + - name: Install nncf-latest + if: ${{ matrix.nncf != 'nncf-develop' }} + run: pip install .[nncf] + - name: Pip freeze run: pip freeze - name: OpenVINO tests - run: | - pip install ${{ matrix.nncf }} - pytest tests/openvino -s --durations=0 + run: pytest tests/openvino -s --durations=0 env: RUN_SLOW: 1 HF_HUB_READ_TOKEN: ${{ secrets.HF_HUB_READ_TOKEN }}