From f36fa7dc1a2d0369615992d8fef2699cd53cb240 Mon Sep 17 00:00:00 2001 From: DenOfEquity <166248528+DenOfEquity@users.noreply.github.com> Date: Sat, 31 Aug 2024 13:05:02 +0100 Subject: [PATCH] fix for sd2 --- modules/sd_vae_taesd.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/modules/sd_vae_taesd.py b/modules/sd_vae_taesd.py index a835548f..a63f6e6a 100644 --- a/modules/sd_vae_taesd.py +++ b/modules/sd_vae_taesd.py @@ -111,10 +111,8 @@ def decoder_model(): model_name = "taef1_decoder.pth" elif shared.sd_model.is_sdxl: model_name = "taesdxl_decoder.pth" - elif shared.sd_model.is_sd1: - model_name = "taesd_decoder.pth" else: - return None + model_name = "taesd_decoder.pth" loaded_model = sd_vae_taesd_models.get(model_name)