From 31c9c751fc1d85c78b9ea89c1c4ce649b61026ad Mon Sep 17 00:00:00 2001 From: Samit <285365963@qq.com> Date: Tue, 7 Jul 2026 17:32:32 +0800 Subject: [PATCH] [Bugfix] Fix LoRA arguments passing in offline text-to-image script (#4936) Signed-off-by: SamitHuang <285365963@qq.com> Co-authored-by: Cursor --- examples/offline_inference/text_to_image/text_to_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/offline_inference/text_to_image/text_to_image.py b/examples/offline_inference/text_to_image/text_to_image.py index a2236d8ca..a8d980c6f 100644 --- a/examples/offline_inference/text_to_image/text_to_image.py +++ b/examples/offline_inference/text_to_image/text_to_image.py @@ -546,8 +546,8 @@ def main(): diffusion_params.extra_args.update({k: v for k, v in user_extra.items() if v is not None}) if lora_request: - diffusion_params.extra_args["lora_request"] = lora_request - diffusion_params.extra_args["lora_scale"] = args.lora_scale + diffusion_params.lora_request = lora_request + diffusion_params.lora_scale = args.lora_scale # Build per-stage sampling params for multi-stage models (e.g. BAGEL), # or wrap single diffusion params for single-stage models.