From 3b7e89d969c02b80ff3662f34813679167b835fa Mon Sep 17 00:00:00 2001 From: VALADI K JAGANATHAN Date: Sat, 6 Apr 2024 00:43:29 +0530 Subject: [PATCH] Update smZ_cfg_denoiser.py --- py/smZ_cfg_denoiser.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/smZ_cfg_denoiser.py b/py/smZ_cfg_denoiser.py index 34189bb..2fc12b4 100644 --- a/py/smZ_cfg_denoiser.py +++ b/py/smZ_cfg_denoiser.py @@ -364,7 +364,7 @@ def calc_cond(c, current_step): return _cond # =========================================================== -class CFGNoisePredictor(CFGNoisePredictorOrig): +class CFGNoisePredictor: def __init__(self, model): super().__init__(model) self.step = 0