From 639e40970f9b88c019866a6babb34f3294465d39 Mon Sep 17 00:00:00 2001 From: "Alex \"mcmonkey\" Goodwin" Date: Sat, 29 Jul 2023 20:47:23 -0700 Subject: [PATCH] fix SDXL compat, for #56 --- scripts/dynamic_thresholding.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/scripts/dynamic_thresholding.py b/scripts/dynamic_thresholding.py index 827a02d..122789b 100644 --- a/scripts/dynamic_thresholding.py +++ b/scripts/dynamic_thresholding.py @@ -170,6 +170,8 @@ class CustomCFGDenoiser(sd_samplers_kdiffusion.CFGDenoiser): self.main_class = dtData def combine_denoised(self, x_out, conds_list, uncond, cond_scale): + if isinstance(uncond, dict) and 'crossattn' in uncond: + uncond = uncond['crossattn'] denoised_uncond = x_out[-uncond.shape[0]:] # conds_list shape is (batch, cond, 2) weights = torch.tensor(conds_list, device=uncond.device).select(2, 1)