VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-18 01:30:02 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add guidance rescale options for Adaptive Projected Guidance in inference

Browse Source
This commit is contained in:
Kohya S
2025-09-21 13:03:14 +09:00
parent 9621d9d637
commit 040d976597
3 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
hunyuan_image_minimal_inference.py
View File

@@ -85,7 +85,13 @@ def parse_args() -> argparse.Namespace:
"--guidance_rescale",
type=float,
default=0.0,
help="Guidance rescale factor for steps without APG, 0.0 to 1.0. Default is 0.0 (no rescale)."
help="Guidance rescale factor for steps without APG, 0.0 to 1.0. Default is 0.0 (no rescale).",
)
parser.add_argument(
"--guidance_rescale_apg",
type=float,
default=0.0,
help="Guidance rescale factor for steps with APG, 0.0 to 1.0. Default is 0.0 (no rescale).",
)
parser.add_argument("--prompt", type=str, default=None, help="prompt for generation")
parser.add_argument("--negative_prompt", type=str, default="", help="negative prompt for generation, default is empty string")
@@ -695,10 +701,18 @@ def generate_body(
# Prepare Guider
cfg_guider_ocr = hunyuan_image_utils.AdaptiveProjectedGuidance(
guidance_scale=10.0, eta=0.0, adaptive_projected_guidance_rescale=10.0, adaptive_projected_guidance_momentum=-0.5
guidance_scale=10.0,
eta=0.0,
adaptive_projected_guidance_rescale=10.0,
adaptive_projected_guidance_momentum=-0.5,
guidance_rescale=args.guidance_rescale_apg,
)
cfg_guider_general = hunyuan_image_utils.AdaptiveProjectedGuidance(
guidance_scale=10.0, eta=0.0, adaptive_projected_guidance_rescale=10.0, adaptive_projected_guidance_momentum=-0.5
guidance_scale=10.0,
eta=0.0,
adaptive_projected_guidance_rescale=10.0,
adaptive_projected_guidance_momentum=-0.5,
guidance_rescale=args.guidance_rescale_apg,
)
# Denoising loop