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