From a2c5893f7f01ba3775d1b29d8deb1d92ef37ccbe Mon Sep 17 00:00:00 2001 From: Vladimir Mandic Date: Sun, 13 Oct 2024 14:23:25 -0400 Subject: [PATCH] fix xyz apply unet Signed-off-by: Vladimir Mandic --- scripts/xyz_grid_classes.py | 4 ++++ scripts/xyz_grid_shared.py | 1 + 2 files changed, 5 insertions(+) diff --git a/scripts/xyz_grid_classes.py b/scripts/xyz_grid_classes.py index db5684f97..8a78d2c40 100644 --- a/scripts/xyz_grid_classes.py +++ b/scripts/xyz_grid_classes.py @@ -36,6 +36,7 @@ class SharedSettingsStackHelper(object): sd_unet = None sd_text_encoder = None extra_networks_default_multiplier = None + disable_weights_auto_swap = None def __enter__(self): #Save overridden settings so they can be restored later. @@ -50,9 +51,12 @@ class SharedSettingsStackHelper(object): self.sd_unet = shared.opts.sd_unet self.sd_text_encoder = shared.opts.sd_text_encoder self.extra_networks_default_multiplier = shared.opts.extra_networks_default_multiplier + self.disable_weights_auto_swap = shared.opts.disable_weights_auto_swap + shared.opts.data["disable_weights_auto_swap"] = False def __exit__(self, exc_type, exc_value, tb): #Restore overriden settings after plot generation. + shared.opts.data["disable_weights_auto_swap"] = self.disable_weights_auto_swap shared.opts.data["sd_vae"] = self.vae shared.opts.data["schedulers_solver_order"] = self.schedulers_solver_order shared.opts.data["tome_ratio"] = self.tome_ratio diff --git a/scripts/xyz_grid_shared.py b/scripts/xyz_grid_shared.py index 963121ddd..b191f58b9 100644 --- a/scripts/xyz_grid_shared.py +++ b/scripts/xyz_grid_shared.py @@ -152,6 +152,7 @@ def apply_unet(p, x, xs): if x == 'None': return p.override_settings['sd_unet'] = x + shared.opts.data['sd_unet'] = x sd_unet.load_unet(shared.sd_model) shared.log.debug(f'XYZ grid apply unet: "{x}"')