From 378ed4567f3290823d5dc5e9556c7d742dc82d23 Mon Sep 17 00:00:00 2001 From: shiimizu Date: Wed, 10 Apr 2024 20:52:00 -0700 Subject: [PATCH] Fix recursion error (part 2). --- smZNodes.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/smZNodes.py b/smZNodes.py index a26a0d0..0ae5773 100644 --- a/smZNodes.py +++ b/smZNodes.py @@ -952,11 +952,9 @@ def get_cond(c, current_step, reverse=False): return (_cond, prompt_editing) try: - comfy.samplers.CFGGuider_orig_smz = comfy.samplers.CFGGuider - CFGGuiderOrig = comfy.samplers.CFGGuider_orig_smz + CFGGuiderOrig = comfy.samplers.CFGGuider_orig_smz = comfy.samplers.CFGGuider except Exception: - comfy.samplers.CFGNoisePredictor_orig_smz = comfy.samplers.CFGNoisePredictor - CFGGuiderOrig = comfy.samplers.CFGNoisePredictor_orig_smz + CFGGuiderOrig = comfy.samplers.CFGGuider_orig_smz = comfy.samplers.CFGNoisePredictor class CFGGuider(CFGGuiderOrig): def __init__(self, model): self.conds = {} @@ -972,10 +970,13 @@ def __init__(self, model): self.sampler = None self.steps_multiplier = 1 - def predict_noise(self, *args, **kwargs): - return self.apply_model(*args, **kwargs) + def __call__(self, *args, **kwargs): + return self.predict_noise(*args, **kwargs) def apply_model(self, *args, **kwargs): + return self.predict_noise(*args, **kwargs) + + def predict_noise(self, *args, **kwargs): if not hasattr(self, 'inner_model2'): self.inner_model2 = CFGDenoiser(self.inner_model.apply_model) x=kwargs['x'] if 'x' in kwargs else args[0]