Fix -m model flag causing 502 on /api/comfyui/generate

Family detection was force-injecting a default VAE (e.g. sdxl_vae.safetensors)
when a model was specified without an explicit --vae. If that VAE file didn't
exist on the ComfyUI server, the workflow was silently rejected. Now only
overrides VAE when the user explicitly passes --vae.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
2026-04-07 16:18:03 +02:00
parent 4a2fdce115
commit b074a7dd50
+3 -2
View File
@@ -260,8 +260,9 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]:
steps = family_defaults["steps"] steps = family_defaults["steps"]
if request.cfg == 7.0: # Default value in schema if request.cfg == 7.0: # Default value in schema
cfg = family_defaults["cfg"] cfg = family_defaults["cfg"]
if request.vae is None: # No VAE specified, use family default # Only override VAE if user explicitly specified one;
vae = family_defaults.get("vae") # None means use checkpoint VAE # otherwise use checkpoint's built-in VAE (vae stays None)
# to avoid failures when a family default VAE file doesn't exist.
logger.debug( logger.debug(
"Detected model family: %s (sampler=%s, scheduler=%s, steps=%d, cfg=%.1f, vae=%s)", "Detected model family: %s (sampler=%s, scheduler=%s, steps=%d, cfg=%.1f, vae=%s)",