From 85eaa305cccd11fa09407d3ed8062902ee1a6207 Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 12 Dec 2023 18:07:56 +0530 Subject: [PATCH 1/2] 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 From 547febafba2117dcda6d142896be1098124c4efd Mon Sep 17 00:00:00 2001 From: cmdr2 Date: Tue, 12 Dec 2023 18:10:04 +0530 Subject: [PATCH 2/2] Autosave the VAE tiling setting --- ui/media/js/auto-save.js | 1 + 1 file changed, 1 insertion(+) diff --git a/ui/media/js/auto-save.js b/ui/media/js/auto-save.js index 1aec8a55..47a2885b 100644 --- a/ui/media/js/auto-save.js +++ b/ui/media/js/auto-save.js @@ -56,6 +56,7 @@ const SETTINGS_IDS_LIST = [ "extract_lora_from_prompt", "embedding-card-size-selector", "lora_model", + "enable_vae_tiling", ] const IGNORE_BY_DEFAULT = ["prompt"]