diff options
| author | AUTOMATIC <16777216c@gmail.com> | 2023-01-16 09:27:52 +0300 |
|---|---|---|
| committer | AUTOMATIC <16777216c@gmail.com> | 2023-01-16 09:27:59 +0300 |
| commit | 972f5785073b8ba5957add72debd74fc56ee9329 (patch) | |
| tree | b8174699512c95d6f3c5be23069ed5a86b7f9752 /scripts/xy_grid.py | |
| parent | dd292a925efc32ad580c927eeeb6e79592833be8 (diff) | |
fix problems related to checkpoint/VAE switching in XY plot
Diffstat (limited to 'scripts/xy_grid.py')
| -rw-r--r-- | scripts/xy_grid.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/scripts/xy_grid.py b/scripts/xy_grid.py index 13a3a046..0cdfa952 100644 --- a/scripts/xy_grid.py +++ b/scripts/xy_grid.py @@ -263,14 +263,12 @@ class SharedSettingsStackHelper(object): def __enter__(self):
self.CLIP_stop_at_last_layers = opts.CLIP_stop_at_last_layers
self.hypernetwork = opts.sd_hypernetwork
- self.model = shared.sd_model
self.vae = opts.sd_vae
def __exit__(self, exc_type, exc_value, tb):
- modules.sd_models.reload_model_weights(self.model)
-
opts.data["sd_vae"] = self.vae
- modules.sd_vae.reload_vae_weights(self.model)
+ modules.sd_models.reload_model_weights()
+ modules.sd_vae.reload_vae_weights()
hypernetwork.load_hypernetwork(self.hypernetwork)
hypernetwork.apply_strength()
|