diff --git a/stg.py b/stg.py
index 9e54821..8f7e079 100644
--- a/stg.py
+++ b/stg.py
@@ -461,7 +461,6 @@ def predict_noise(
                 stg_result,
                 noise_pred_neg,
                 cfg_scale=self.apg_cfg_scale,
-                momentum_buffer=None,
                 eta=self.eta,
                 norm_threshold=self.norm_threshold,
             )