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

microsoft / Olive

Olive: Simplify ML Model Finetuning, Conversion, Quantization, and Optimization for CPUs, GPUs and NPUs.

https://microsoft.github.io/Olive/

MIT License

1.58k stars 165 forks source link

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

Open peterer0625 opened 5 months ago

peterer0625 commented 5 months ago

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.

Jay19751103 commented 2 months ago

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?