Update sd_models.py

This commit is contained in:
lllyasviel
2024-01-25 07:18:15 -08:00
parent f993ab1c2c
commit 232c90aba2

View File

@@ -522,14 +522,7 @@ class SdModelData:
sd_vae.base_vae = getattr(v, "base_vae", None)
sd_vae.loaded_vae_file = getattr(v, "loaded_vae_file", None)
sd_vae.checkpoint_info = v.sd_checkpoint_info
try:
self.loaded_sd_models.remove(v)
except ValueError:
pass
if v is not None:
self.loaded_sd_models.insert(0, v)
self.loaded_sd_models = [self.sd_model]
model_data = SdModelData()