优化全能创作

2026-09-29 15:15:54 +08:00
parent 1130ecd4ba
commit c687ec4bbd
16 changed files with 632 additions and 258 deletions
@@ -3980,7 +3980,15 @@ def run_standalone_image_task(*, task_id: str) -> None:
            vsize = "2304x1728" if payload.get("reference_product") else _ratio_to_volcano_size(output_ratio)
            response = provider.image_generation(model=model_config.name, endpoint=model_config.endpoint, prompt=edit_prompt, image=edit_images, size=vsize)
        else:
-            response = provider.image_generation(model=model_config.name, endpoint=model_config.endpoint, prompt=gen_prompt)
+            generate_kwargs = {"model": model_config.name, "endpoint": model_config.endpoint, "prompt": gen_prompt}
+            if mode == "model" and payload.get("feature") == "omni_create" and output_ratio == "16:9":
+                # 全能创作角色设定图没有参考图，旧分支漏传 size，会退回供应商默认竖幅。
+                generate_kwargs["size"] = (
+                    _ratio_to_volcano_size(output_ratio)
+                    if model_config.provider.name in OFFICIAL_DIRECT_PROVIDERS
+                    else _ratio_to_image_size(output_ratio)
+                )
+            response = provider.image_generation(**generate_kwargs)
        if not use_model_routing:
            media = provider.extract_first_media_url(response)
        with transaction.atomic():