diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 13:23:30 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2022-10-09 13:23:30 +0300 |
| commit | f4578b343ded3b8ccd1879ea0c0b3cdadfcc3a5f (patch) | |
| tree | 42089b10ce6e431d5fc4ba2dd0bb38f9ab7acd45 /modules/sd_models.py | |
| parent | bd833409ac7b8337040d521f6b65ced51e1b2ea8 (diff) | |
fix model switching not working properly if there is a different yaml config
Diffstat (limited to 'modules/sd_models.py')
| -rw-r--r-- | modules/sd_models.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/modules/sd_models.py b/modules/sd_models.py index 2101b18d..d0c74dd8 100644 --- a/modules/sd_models.py +++ b/modules/sd_models.py @@ -196,7 +196,8 @@ def reload_model_weights(sd_model, info=None): return
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:
lowvram.send_everything_to_cpu()
|