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

This commit is contained in:
Changyi Yang
2026-02-23 02:19:53 -05:00
committed by GitHub
parent 6a999dbdf8
commit c3c1532820

View File

@@ -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 (