minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] feat: detect Flux2 custom VAE path from component_paths (#19170)

Browse Source
This commit is contained in:
Changyi Yang
2026-02-23 15:19:53 +08:00
committed by GitHub
parent 6a999dbdf8
commit c3c1532820
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/configs/pipeline_configs/base.py
+6
View File
@@ -599,6 +599,12 @@ class PipelineConfig:
# 1.5. Adjust pipeline config for fine-tuned VAE if needed # 1.5. Adjust pipeline config for fine-tuned VAE if needed
pipeline_config_cls = model_info.pipeline_config_cls pipeline_config_cls = model_info.pipeline_config_cls
vae_path = kwargs.get(prefix_with_dot + "vae_path") or kwargs.get("vae_path") 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 # Check if this is a Flux2 model with fal/FLUX.2-Tiny-AutoEncoder
if ( if (