From 85eaa305cccd11fa09407d3ed8062902ee1a6207 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 12 Dec 2023 18:07:56 +0530 Subject: [PATCH] Hotfix for #1701 - run disable VAE tiling only on pipelines that support it --- ui/easydiffusion/tasks/render_images.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/ui/easydiffusion/tasks/render_images.py b/ui/easydiffusion/tasks/render_images.py index 50c76c2e..7d8fbc5e 100644 --- a/ui/easydiffusion/tasks/render_images.py +++ b/ui/easydiffusion/tasks/render_images.py @@ -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