[diffusion] fix: resolve library mismatch in scheduler and update dit offload method name (#17916)

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]

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