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 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"]