Skip to content

cfg effect experiment #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kevin-on wants to merge 8 commits into
base: main
Choose a base branch
from
Open

cfg effect experiment #5

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
ff24872
Add CFG effect experiment implementation
kevin-on Dec 11, 2025
78e0cb2
Add gradient norm tracking to CFG effect experiments
kevin-on Dec 11, 2025
4cf5c32
Refactor IntermediateImageSaver and GradientNormTracker to improve ou…
kevin-on Dec 12, 2025
b5ef4fa
Merge branch 'main' into kevin/cfg-experiment
kevin-on Dec 12, 2025
722695a
Update IntermediateImageSaver instantiation to use 'output_folder' pa…
kevin-on Dec 12, 2025
acfd9fd
Refactor '--track-grad-norm' argument to use BooleanOptionalAction ac…
kevin-on Dec 12, 2025
c84d146
Refactor sampling utilities to consolidate hooks into a new module
kevin-on Dec 12, 2025
9acff90
Enhance EqM forward pass and gradient norm tracking for CFG
kevin-on Dec 12, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 6 additions & 3 deletions deprecated/sample_eqm_two.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -77,7 +77,7 @@ def main(args):
save_steps_list = []
if args.save_steps is not None:
save_steps_list = [int(s.strip()) for s in args.save_steps.split(",")]
img_saver = IntermediateImageSaver(save_steps_list, args.out)
img_saver = IntermediateImageSaver(save_steps_list, output_folder=args.out)
hooks.append(img_saver)
print(f"Created IntermediateImageSaver hook for steps: {save_steps_list}")

Expand Down Expand Up @@ -117,7 +117,7 @@ def main(args):
# Finalize gradient norm statistics if enabled
if grad_tracker is not None:
print("Computing gradient norm statistics...")
grad_tracker.finalize(args, args.out)
grad_tracker.finalize(args.out, args.num_sampling_steps, args.stepsize, args.sampler)

# Create .npz files for FID evaluation
print("Creating .npz file for final samples...")
Expand Down Expand Up @@ -168,7 +168,10 @@ def main(args):
help="Comma-separated list of sampling steps to save intermediate images (e.g., '0,50,100,249')",
)
parser.add_argument(
"--track-grad-norm", action="store_true", help="Enable gradient norm tracking and visualization"
"--track-grad-norm",
action=argparse.BooleanOptionalAction,
default=True,
help="Enable gradient norm tracking and visualization",
)
parser.add_argument("--uncond", type=bool, default=True, help="Disable/enable noise conditioning (default: True)")
parser.add_argument(
Expand Down
Loading