From 39458ec0e3b938fb5f21c1769a4bde046ed924c9 Mon Sep 17 00:00:00 2001 From: kohya-ss <52813779+kohya-ss@users.noreply.github.com> Date: Tue, 16 Sep 2025 21:17:21 +0900 Subject: [PATCH] fix: update default values for guidance_scale, image_size, infer_steps, and flow_shift in argument parser --- hunyuan_image_minimal_inference.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/hunyuan_image_minimal_inference.py b/hunyuan_image_minimal_inference.py index 7db490cd..04ab1aac 100644 --- a/hunyuan_image_minimal_inference.py +++ b/hunyuan_image_minimal_inference.py @@ -67,12 +67,12 @@ def parse_args() -> argparse.Namespace: # inference parser.add_argument( - "--guidance_scale", type=float, default=5.0, help="Guidance scale for classifier free guidance. Default is 5.0." + "--guidance_scale", type=float, default=3.5, help="Guidance scale for classifier free guidance. Default is 3.5." ) 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") - parser.add_argument("--image_size", type=int, nargs=2, default=[256, 256], help="image size, height and width") - parser.add_argument("--infer_steps", type=int, default=25, help="number of inference steps, default is 25") + parser.add_argument("--image_size", type=int, nargs=2, default=[2048, 2048], help="image size, height and width") + parser.add_argument("--infer_steps", type=int, default=50, help="number of inference steps, default is 50") parser.add_argument("--save_path", type=str, required=True, help="path to save generated video") parser.add_argument("--seed", type=int, default=None, help="Seed for evaluation.") @@ -80,8 +80,8 @@ def parse_args() -> argparse.Namespace: parser.add_argument( "--flow_shift", type=float, - default=None, - help="Shift factor for flow matching schedulers. Default is None (default).", + default=5.0, + help="Shift factor for flow matching schedulers. Default is 5.0.", ) parser.add_argument("--fp8", action="store_true", help="use fp8 for DiT model")