Update smZ_cfg_denoiser.py

This commit is contained in:
VALADI K JAGANATHAN
2024-04-06 00:43:29 +05:30
committed by GitHub
parent a5fe47c531
commit 3b7e89d969

View File

@@ -364,7 +364,7 @@ def calc_cond(c, current_step):
return _cond return _cond
# =========================================================== # ===========================================================
class CFGNoisePredictor(CFGNoisePredictorOrig): class CFGNoisePredictor:
def __init__(self, model): def __init__(self, model):
super().__init__(model) super().__init__(model)
self.step = 0 self.step = 0