[Bug]: Need to reload web UI for each new prompt/batch of images

[slumberspirit]

What happened?

While using the public URL link generated, when I first generate a batch of images and it finishes, it displays them on that public URL page. Every subsequent batch doesn't display them. The number of seconds just keeps ticking up even when collab tells me it's done. Then I have to reload the public URL and re-enter my settings to run another prompt.

Version

0.0.1 (Default)

What browsers are you seeing the problem on?

Firefox

Where are you running the webui?

Colab

Custom settings

No response

Relevant log output

No response

Code of Conduct

• I agree to follow this project's Code of Conduct

[Lyntai]

The same issue happening to me, the generated images show only with batch 1.

[EliteRaptoR]

Same problem here. If you need more info tell me.

[rustyorb]

I am also having the same issue, on colab pro 'HIGH RAM' set. Sometimes I may get more than one run, but not often any more. Seems like it's gotten worst the last day or so for some reason.

[wulfenbach-jpg]

Same issue here. Colab Pro, capable of doing more than one run occasionally, however, most of the time the colab has finished generation whilst the webui continues to spin and count, well into the hundreds of seconds without showing me the images which are present in the google drive folder. Was very rare yesterday but seems to have suddenly appeared nearly every single generation in the last day.

[rustyorb]

I wanted to add that if I switch the sampler to DDIM it stays consistent. Meaning I don't have to reload/restart.

Addition. For me, K_LMS is the only problem. If I have that chosen, I have issue, other samplers work fine.

[EliteRaptoR]

When DDIM is selected I can run 4 to 8 images (1 image each bach) before the problem appear.
So, in my case seems that the problem is not only in K_LMS.