diff --git a/python/sglang/multimodal_gen/runtime/loader/component_loader.py b/python/sglang/multimodal_gen/runtime/loader/component_loader.py index 71a710075..bed503b9d 100644 --- a/python/sglang/multimodal_gen/runtime/loader/component_loader.py +++ b/python/sglang/multimodal_gen/runtime/loader/component_loader.py @@ -292,6 +292,11 @@ class ComponentLoader(ABC): "video_dit", ]: transformers_or_diffusers = "diffusers" + if ( + module_type == "scheduler" + and transformers_or_diffusers == "mova.diffusion.schedulers.flow_match_pair" + ): + transformers_or_diffusers = "diffusers" if module_type in module_loaders: loader_cls, expected_library = module_loaders[module_type] diff --git a/python/sglang/multimodal_gen/runtime/models/model_stages/mova.py b/python/sglang/multimodal_gen/runtime/models/model_stages/mova.py index afbe4a914..e83336766 100644 --- a/python/sglang/multimodal_gen/runtime/models/model_stages/mova.py +++ b/python/sglang/multimodal_gen/runtime/models/model_stages/mova.py @@ -566,7 +566,7 @@ class MOVADenoisingStage(PipelineStage): for dit in filter(None, [self.video_dit, self.video_dit_2, self.audio_dit]): if isinstance(dit, OffloadableDiTMixin): - dit.prepare_for_next_denoise() + dit.prepare_for_next_req() return batch