fix model switching not working properly if there is a different yaml config

This commit is contained in:
AUTOMATIC 2022-10-09 13:23:30 +03:00
parent bd833409ac
commit f4578b343d
1 changed files with 2 additions and 1 deletions

View File

@ -196,7 +196,8 @@ def reload_model_weights(sd_model, info=None):
return return
if sd_model.sd_checkpoint_info.config != checkpoint_info.config: if sd_model.sd_checkpoint_info.config != checkpoint_info.config:
return load_model() shared.sd_model = load_model()
return shared.sd_model
if shared.cmd_opts.lowvram or shared.cmd_opts.medvram: if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
lowvram.send_everything_to_cpu() lowvram.send_everything_to_cpu()