From c3c15328207b798fbedf0f46d6c8466f7225b5c0 Mon Sep 17 00:00:00 2001
From: Changyi Yang <112288487+ChangyiYang@users.noreply.github.com>
Date: Mon, 23 Feb 2026 02:19:53 -0500
Subject: [PATCH] [diffusion] feat: detect Flux2 custom VAE path from
 component_paths (#19170)

---
 .../sglang/multimodal_gen/configs/pipeline_configs/base.py  | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/python/sglang/multimodal_gen/configs/pipeline_configs/base.py b/python/sglang/multimodal_gen/configs/pipeline_configs/base.py
index b220029a8..5c0989045 100644
--- a/python/sglang/multimodal_gen/configs/pipeline_configs/base.py
+++ b/python/sglang/multimodal_gen/configs/pipeline_configs/base.py
@@ -599,6 +599,12 @@ class PipelineConfig:
             # 1.5. Adjust pipeline config for fine-tuned VAE if needed
             pipeline_config_cls = model_info.pipeline_config_cls
         vae_path = kwargs.get(prefix_with_dot + "vae_path") or kwargs.get("vae_path")
+        if vae_path is None:
+            component_paths = kwargs.get(
+                prefix_with_dot + "component_paths"
+            ) or kwargs.get("component_paths")
+            if isinstance(component_paths, dict):
+                vae_path = component_paths.get("vae")
 
         # Check if this is a Flux2 model with fal/FLUX.2-Tiny-AutoEncoder
         if (