fix problems related to checkpoint/VAE switching in XY plot

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()