microsoft/Olive

"num_images" doesn't work for the example of directml stable_diffusion_xl.

peterer0625 opened this issue · 1 comments

Describe the bug
"num_images" doesn't work for the example of directml stable_diffusion_xl.

To Reproduce
Command: python stable_diffusion_xl.py --num_image 2 --prompt="great wall" --num_inference_steps=4

Expected behavior
Generate two images. However, I only saw one image.

hi Olive Team
It looks that while images_saved < num_images: removed in run_inference_loop
or just use batch_size to replace it on SDXL?
anyone can fix this?