holm/easydiffusion
1
0
Fork 0
forked from extern/easydiffusion
Code Issues Pull Requests 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") delattr(pipe.unet, "_allocate_trt_buffers")
if task_data.enable_vae_tiling: if task_data.enable_vae_tiling:
pipe.enable_vae_tiling() if hasattr(pipe, "enable_vae_tiling"):
pipe.enable_vae_tiling()
else: else:
pipe.disable_vae_tiling() if hasattr(pipe, "disable_vae_tiling"):
pipe.disable_vae_tiling()
images = generate_images(context, callback=callback, **req.dict()) images = generate_images(context, callback=callback, **req.dict())
user_stopped = False user_stopped = False