fix for sd2

This commit is contained in:
DenOfEquity 2024-08-31 13:05:02 +01:00 committed by GitHub
parent 7ffd124a9e
commit f36fa7dc1a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)