From 3f9eab49467ba2d224d48464aac11cb07b85dbb1 Mon Sep 17 00:00:00 2001 From: Kohya S Date: Wed, 9 Jul 2025 23:33:50 +0900 Subject: [PATCH] fix: update default values in lumina minimal inference as same as sample generation --- lumina_minimal_inference.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lumina_minimal_inference.py b/lumina_minimal_inference.py index ba305f6f..4f915179 100644 --- a/lumina_minimal_inference.py +++ b/lumina_minimal_inference.py @@ -205,8 +205,8 @@ def setup_parser() -> argparse.ArgumentParser: parser.add_argument("--negative_prompt", type=str, default="", help="Negative prompt for image generation, default is empty") parser.add_argument("--output_dir", type=str, default="outputs", help="Output directory for generated images") parser.add_argument("--seed", type=int, default=None, help="Random seed") - parser.add_argument("--steps", type=int, default=30, help="Number of inference steps") - parser.add_argument("--guidance_scale", type=float, default=4.0, help="Guidance scale for classifier-free guidance") + parser.add_argument("--steps", type=int, default=36, help="Number of inference steps") + parser.add_argument("--guidance_scale", type=float, default=3.5, help="Guidance scale for classifier-free guidance") parser.add_argument("--image_width", type=int, default=1024, help="Image width") parser.add_argument("--image_height", type=int, default=1024, help="Image height") parser.add_argument("--dtype", type=str, default="bf16", help="Data type for model (bf16, fp16, float)") @@ -224,7 +224,7 @@ def setup_parser() -> argparse.ArgumentParser: parser.add_argument( "--discrete_flow_shift", type=float, - default=1.0, + default=6.0, help="Shift value for FlowMatchEulerDiscreteScheduler", ) parser.add_argument(