stable-diffusion-webui
stable-diffusion-webui copied to clipboard
[Feature Request]: parallelize batch upscaling
Is there an existing issue for this?
- [X] I have searched the existing issues and checked the recent builds/commits
What would your feature do ?
Upscale huge batches of images faster by feeding multiple images to the upscaler nets such as esrgan or swinrl in increased batch sizes directly.
Right now, the upscalers process only a single image at the time and it's a huge loss of time accumulating for large sequences, for example, Deforum's output videos.
For comparison, upscalers eta for ~1 min Deforum videos reaches around 1h, meanwhile Topaz video (where upscalers are obviously parallelized, if not run on multiple gpus) has only ~5 mins.
Proposed workflow
- Go to extras/launch upscalers from an extension using scripts api
- Select desired params.
- Select the batch size that fits in your GPU. (esp with SD unloaded from memory)
- Press generate
- ...
- Profit! The process should be much faster now
Additional information
No response