Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

create_image_gen_model for PT framework does not return callback #1310

Merged
merged 1 commit into from
Dec 5, 2024

Conversation

anzr299
Copy link
Contributor

@anzr299 anzr299 commented Dec 5, 2024

When trying to benchmark a Stable diffusion 3 model with pt framework, the create_image_gen_model does not return the expected callback in image generation task

Error Traceback:

[ ERROR ] An exception occurred
[ INFO ] Traceback (most recent call last):
  File "openvino.genai/tools/llm_bench/benchmark.py", line 211, in main
    iter_data_list, pretrain_time, iter_timestamp = CASE_TO_BENCH[model_args['use_case']](
  File "openvino.genai/tools/llm_bench/task/image_generation.py", line 171, in run_image_generation_benchmark
    pipe, pretrain_time, use_genai, callback = FW_UTILS[framework].create_image_gen_model(model_path, device, **args)
ValueError: not enough values to unpack (expected 4, got 3)

@github-actions github-actions bot added the category: llm_bench Label for tool/llm_bench folder label Dec 5, 2024
@anzr299 anzr299 requested a review from eaidova December 5, 2024 11:41
@eaidova eaidova merged commit 3a1bd24 into master Dec 5, 2024
48 of 54 checks passed
@ilya-lavrenov ilya-lavrenov deleted the anzr299-patch-1 branch December 6, 2024 07:44
sungeunk pushed a commit to sungeunk/openvino.genai that referenced this pull request Dec 16, 2024
…penvinotoolkit#1310)

When trying to benchmark a Stable diffusion 3 model with pt framework,
the `create_image_gen_model` does not return the expected callback in
[image generation
task](https://github.com/openvinotoolkit/openvino.genai/blob/d294db9469c4795bed24ebfb1318cc68adc682e0/tools/llm_bench/task/image_generation.py#L168)

Error Traceback:
```
[ ERROR ] An exception occurred
[ INFO ] Traceback (most recent call last):
  File "openvino.genai/tools/llm_bench/benchmark.py", line 211, in main
    iter_data_list, pretrain_time, iter_timestamp = CASE_TO_BENCH[model_args['use_case']](
  File "openvino.genai/tools/llm_bench/task/image_generation.py", line 171, in run_image_generation_benchmark
    pipe, pretrain_time, use_genai, callback = FW_UTILS[framework].create_image_gen_model(model_path, device, **args)
ValueError: not enough values to unpack (expected 4, got 3)
```
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
category: llm_bench Label for tool/llm_bench folder
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants