omerbt / MultiDiffusion

Official Pytorch Implementation for "MultiDiffusion: Fusing Diffusion Paths for Controlled Image Generation" presenting "MultiDiffusion" (ICML 2023)
https://multidiffusion.github.io/
979 stars 57 forks source link

Bug in vae_optimize.py, vae_tile_forward uses 'result' when result is None #16

Closed skfoo closed 1 year ago

skfoo commented 1 year ago

(Updated) The last line of vae_tile_forward (vae_optimize.py line 650) is

return result if result is not None else result_approx.to(device)

When you interrupt the generation using HiRes fix (click Interrupt in the webui), an exception is thrown from this line,

     File "stable-diffusion-webui/extensions/multidiffusion-upscaler-for-automatic1111/scripts/vae_optimize.py", line 650, in vae_tile_forward
        return result if result is not None else result_approx.to(device)
    AttributeError: 'NoneType' object has no attribute 'to'
skfoo commented 1 year ago

Wrong repo (-‸ლ)