extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 00:03:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Hotfix for #1701 - run disable VAE tiling only on pipelines that support it

Browse Source
This commit is contained in:
cmdr2 2023-12-12 18:07:56 +05:30
parent e77629c525
commit 85eaa305cc
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ui/easydiffusion/tasks/render_images.py
View File

@ -279,9 +279,11 @@ def generate_images_internal(
delattr(pipe.unet, "_allocate_trt_buffers")
if task_data.enable_vae_tiling:
pipe.enable_vae_tiling()
if hasattr(pipe, "enable_vae_tiling"):
pipe.enable_vae_tiling()
else:
pipe.disable_vae_tiling()
if hasattr(pipe, "disable_vae_tiling"):
pipe.disable_vae_tiling()
images = generate_images(context, callback=callback, **req.dict())
user_stopped = False