Skip to content

Commit 99043f3

Browse files
committed
fix one of previous merges breaking the program
1 parent c250d2a commit 99043f3

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

modules/sd_models.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -167,6 +167,8 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"):
167167
sd_vae.restore_base_vae(model)
168168
checkpoints_loaded[model.sd_checkpoint_info] = model.state_dict().copy()
169169

170+
vae_file = sd_vae.resolve_vae(checkpoint_file, vae_file=vae_file)
171+
170172
if checkpoint_info not in checkpoints_loaded:
171173
print(f"Loading weights [{sd_model_hash}] from {checkpoint_file}")
172174

0 commit comments

Comments
 (0)