From b074a7dd506e120e259b59ec3e6518e3ca176ee8 Mon Sep 17 00:00:00 2001 From: aladac Date: Tue, 7 Apr 2026 16:18:03 +0200 Subject: [PATCH] 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) --- tensors/server/comfyui_api_routes.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/tensors/server/comfyui_api_routes.py b/tensors/server/comfyui_api_routes.py index 3be8ce7..7e30961 100644 --- a/tensors/server/comfyui_api_routes.py +++ b/tensors/server/comfyui_api_routes.py @@ -260,8 +260,9 @@ def comfyui_generate(request: GenerateRequest) -> dict[str, Any]: steps = family_defaults["steps"] if request.cfg == 7.0: # Default value in schema cfg = family_defaults["cfg"] - if request.vae is None: # No VAE specified, use family default - vae = family_defaults.get("vae") # None means use checkpoint VAE + # Only override VAE if user explicitly specified one; + # otherwise use checkpoint's built-in VAE (vae stays None) + # to avoid failures when a family default VAE file doesn't exist. logger.debug( "Detected model family: %s (sampler=%s, scheduler=%s, steps=%d, cfg=%.1f, vae=%s)",