Skip to content

Commit

Permalink
Fixed WWB im2im tests (#1482)
Browse files Browse the repository at this point in the history
Using `datasets==3.2.0` breaks the Python process with some GIL issue:

```
Fatal Python error: PyGILState_Release: thread state 0x7fa56809ade0 must be current when releasing
Python runtime state: finalizing (tstate=0x0000000000ad2958)

Thread 0x00007fa6ae07c740 (most recent call first):
  <no Python frame>

Extension modules: numpy.core._multiarray_umath, numpy.core._multiarray_tests, numpy.linalg._umath_linalg, numpy.fft._pocketfft_internal, numpy.random._common, numpy.random.bit_generator, numpy.random._boun
ded_integers, numpy.random._mt19937, numpy.random.mtrand, numpy.random._philox, 
```

---------

Co-authored-by: Ekaterina Aidova <ekaterina.aidova@intel.com>
  • Loading branch information
AlexKoff88 and eaidova authored Jan 6, 2025
1 parent 91ebc25 commit d3d628a
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 3 deletions.
1 change: 1 addition & 0 deletions .github/workflows/genai-tools.yml
Original file line number Diff line number Diff line change
Expand Up @@ -181,6 +181,7 @@ jobs:
python -m pip install ${{ env.SRC_DIR }} -v ${{ needs.openvino_download.outputs.ov_wheel_source }}
python -m pip install -r ${{ env.WWB_PATH }}/requirements.txt ${{ needs.openvino_download.outputs.ov_wheel_source }}
python -m pip install git+https://github.com/huggingface/optimum-intel.git@main#egg=optimum-intel
pip install -U --force-reinstall datasets==3.1.0
working-directory: ${{ env.OV_INSTALL_DIR }}
- name: WWB Tests
run: |
Expand Down
1 change: 1 addition & 0 deletions tools/who_what_benchmark/requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,4 @@ pandas>=2.0.3
numpy>=1.23.5
tqdm>=4.66.1
diffusers
datasets<3.2.0
6 changes: 3 additions & 3 deletions tools/who_what_benchmark/tests/test_cli_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -42,8 +42,8 @@ def teardown_module():
("hf-internal-testing/tiny-stable-diffusion-torch", "text-to-image", "hf"),
("hf-internal-testing/tiny-stable-diffusion-torch", "text-to-image", "openvino"),
("hf-internal-testing/tiny-stable-diffusion-xl-pipe", "text-to-image", "hf"),
# ("hf-internal-testing/tiny-stable-diffusion-torch", "image-inpainting", "hf"),
# ("hf-internal-testing/tiny-stable-diffusion-xl-pipe", "image-inpainting", "hf"),
("hf-internal-testing/tiny-stable-diffusion-torch", "image-inpainting", "hf"),
("hf-internal-testing/tiny-stable-diffusion-xl-pipe", "image-inpainting", "hf"),
],
)
def test_image_model_types(model_id, model_type, backend):
Expand Down Expand Up @@ -90,7 +90,7 @@ def test_image_model_types(model_id, model_type, backend):
list(itertools.product(OV_IMAGE_MODELS,
["image-to-image",
"text-to-image",
# "image-inpainting"
"image-inpainting"
])),
)
def test_image_model_genai(model_id, model_type):
Expand Down

0 comments on commit d3d628a

Please sign in to comment.