fix model switching not working properly if there is a different yaml config
This commit is contained in:
parent
bd833409ac
commit
f4578b343d
1 changed files with 2 additions and 1 deletions
|
@ -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()
|
||||||
|
|
Loading…
Reference in a new issue