[diffusion] fix: fix missing component names for VAELoader (#18069)

2026-02-02 09:48:17 +08:00
parent 977096ae03
commit c84cd4b5ff
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/multimodal_gen/runtime/loader/vae_loader.py
+++ b/python/sglang/multimodal_gen/runtime/loader/vae_loader.py
@@ -24,7 +24,7 @@ logger = init_logger(__name__)
 class VAELoader(ComponentLoader):
    """Shared loader for (video/audio) VAE modules."""

-    component_names = ["vae", "audio_vae"]
+    component_names = ["vae", "audio_vae", "video_vae"]
    expected_library = "diffusers"

    def should_offload(