[WWB] add more models

openvinotoolkit · Dec 27, 2024 · 75301ab · 75301ab
1 parent 82b44fa
commit 75301ab
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/.github/workflows/llm_bench-python.yml b/.github/workflows/llm_bench-python.yml
@@ -151,8 +151,8 @@ jobs:
           rm -rf ./ov_models/internvl2-1B
       - name: WWB Tests
         run: |
-          pip install git+https://github.com/huggingface/optimum-intel.git
-          GIT_CLONE_PROTECTION_ACTIVE=false PIP_PRE=1 PIP_EXTRA_INDEX_URL=https://storage.openvinotoolkit.org/simple/wheels/nightly pip install ${{ env.WWB_PATH }}
+          python -m pip install git+https://github.com/huggingface/optimum-intel.git
+          python -m pip install ${{ env.WWB_PATH }}
           python -m pytest -v ${{ env.WWB_PATH }}/tests
   stateful:
     defaults:
@@ -189,9 +189,9 @@ jobs:
         working-directory: ${{ env.OV_INSTALL_DIR }}
       - name: WWB Tests
         run: |
-          pip install pytest
-          pip install git+https://github.com/huggingface/optimum-intel.git
-          GIT_CLONE_PROTECTION_ACTIVE=false PIP_PRE=1 PIP_EXTRA_INDEX_URL=https://storage.openvinotoolkit.org/simple/wheels/nightly pip install ${{ env.WWB_PATH }}
+          python -m pip install pytest
+          python -m pip install git+https://github.com/huggingface/optimum-intel.git
+          python -m pip install ${{ env.WWB_PATH }}
           python -m pytest -v ${{ env.WWB_PATH }}/tests
 
   Overall_Status:

diff --git a/tools/who_what_benchmark/tests/test_cli_image.py b/tools/who_what_benchmark/tests/test_cli_image.py
@@ -65,7 +65,10 @@ def test_image_model_types(model_id, model_type, backend):
 @pytest.mark.parametrize(
     ("model_id", "model_type"),
     [
+        ("echarlaix/tiny-random-latent-consistency", "text-to-image"),
         ("echarlaix/tiny-random-stable-diffusion-xl", "text-to-image"),
+        ("yujiepan/stable-diffusion-3-tiny-random", "text-to-image"),
+        ("katuni4ka/tiny-random-flux", "text-to-image"),
     ],
 )
 def test_image_model_genai(model_id, model_type):