Skip to content

Commit c84cd4b

Browse files
authored
[diffusion] fix: fix missing component names for VAELoader (sgl-project#18069)
1 parent 977096a commit c84cd4b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/sglang/multimodal_gen/runtime/loader/vae_loader.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -24,7 +24,7 @@
2424
class VAELoader(ComponentLoader):
2525
"""Shared loader for (video/audio) VAE modules."""
2626

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

3030
def should_offload(

0 commit comments

Comments
 (0)