From ca82c59250a47e79649d3fb80289436391e41520 Mon Sep 17 00:00:00 2001
From: Omri SirComp <omribz156@gmail.com>
Date: Mon, 25 May 2026 12:24:36 +0300
Subject: [PATCH] fix: remove unsupported APG momentum buffer arg

---
 stg.py | 1 -
 1 file changed, 1 deletion(-)

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,
             )