From 972f5785073b8ba5957add72debd74fc56ee9329 Mon Sep 17 00:00:00 2001
From: AUTOMATIC <16777216c@gmail.com>
Date: Mon, 16 Jan 2023 09:27:52 +0300
Subject: [PATCH] fix problems related to checkpoint/VAE switching in XY plot

---
 scripts/xy_grid.py | 6 ++----
 1 file 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()