Skip to content

fix: remove unsupported APG momentum buffer argument#500

Open
omribz156 wants to merge 1 commit into
Lightricks:masterfrom
omribz156:codex/apg-momentum-buffer
Open

fix: remove unsupported APG momentum buffer argument#500
omribz156 wants to merge 1 commit into
Lightricks:masterfrom
omribz156:codex/apg-momentum-buffer

Conversation

@omribz156
Copy link
Copy Markdown

Summary

Removes the unsupported momentum_buffer=None keyword from the STGGuiderAdvanced APG call.

apg() accepts noise_pred_pos, noise_pred_neg, cfg_scale, eta, and norm_threshold; passing momentum_buffer causes apply_apg=True sampling to fail immediately with the TypeError reported in #492.

Verification

  • python -m py_compile stg.py
  • git diff --check

I did not run a full ComfyUI sampling workflow locally because this environment does not have the ComfyUI runtime/model stack installed.

Implemented with Codex assistance; I kept the patch focused and manually checked the reported call/signature mismatch.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant