fix: do not reload model when VAE stays the same

This commit is contained in:
Manuel Schmid
2024-05-04 21:12:25 +02:00
parent 15696da9b8
commit 4a3aac09a3
3 changed files with 11 additions and 9 deletions
+1 -1
View File
@@ -71,7 +71,7 @@ def refresh_base_model(name, vae_name=None):
if model_base.filename == filename and model_base.vae_filename == vae_filename:
return
model_base = core.StableDiffusionModel()
model_base = core.StableDiffusionModel(vae_filename=vae_filename)
model_base = core.load_model(filename, vae_filename)
print(f'Base model loaded: {model_base.filename}')
return