提交 99043f33 编写于 作者: A AUTOMATIC

fix one of previous merges breaking the program

上级 c250d2a0
......@@ -167,6 +167,8 @@ def load_model_weights(model, checkpoint_info, vae_file="auto"):
sd_vae.restore_base_vae(model)
checkpoints_loaded[model.sd_checkpoint_info] = model.state_dict().copy()
vae_file = sd_vae.resolve_vae(checkpoint_file, vae_file=vae_file)
if checkpoint_info not in checkpoints_loaded:
print(f"Loading weights [{sd_model_hash}] from {checkpoint_file}")
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册