Update README.md

README.md

@@ -161,20 +161,6 @@ from diffusers import (
 from diffusers.utils import export_to_video
 
 
-# DPMSolver++ subclass: ignores pipeline-supplied sigmas and builds its own
-# flow-matching schedule. Will be unnecessary once PR #13551 adds the
-# _is_flow_multistep branch.
-class FlowDPMSolver(DPMSolverMultistepScheduler):
-    def set_timesteps(self, num_inference_steps=None, device=None,
-                      sigmas=None, mu=None, timesteps=None):
-        if sigmas is not None and num_inference_steps is None:
-            num_inference_steps = len(sigmas)
-        super().set_timesteps(
-            num_inference_steps=num_inference_steps,
-            device=device, timesteps=timesteps,
-        )
-
-
 guider = AdaptiveProjectedGuidance(
     guidance_scale=8.0,
     adaptive_projected_guidance_rescale=12.0,
@@ -188,16 +174,6 @@ pipe = MotifVideoPipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     guider=guider,
 )
-
-# DPMSolver++ for faster convergence
-pipe.scheduler = FlowDPMSolver(
-    num_train_timesteps=pipe.scheduler.config.get("num_train_timesteps", 1000),
-    algorithm_type="dpmsolver++",
-    solver_order=2,
-    prediction_type="flow_prediction",
-    use_flow_sigmas=True,
-    flow_shift=15.0,
-)
 pipe = pipe.to("cuda")
 
 output = pipe(
@@ -237,15 +213,6 @@ pipe = MotifVideoImage2VideoPipeline.from_pretrained(
     torch_dtype=torch.bfloat16,
     guider=guider,
 )
-
-pipe.scheduler = FlowDPMSolver(
-    num_train_timesteps=pipe.scheduler.config.get("num_train_timesteps", 1000),
-    algorithm_type="dpmsolver++",
-    solver_order=2,
-    prediction_type="flow_prediction",
-    use_flow_sigmas=True,
-    flow_shift=15.0,
-)
 pipe = pipe.to("cuda")
 
 image = load_image("https://huggingface.co/Motif-Technologies/Motif-Video-2B/resolve/main/assets/i2v_sample.jpg")

scheduler/scheduler_config.json

@@ -1,18 +1,10 @@
 {
-  "_class_name": "FlowMatchEulerDiscreteScheduler",
+  "_class_name": "DPMSolverMultistepScheduler",
   "_diffusers_version": "0.36.0",
-  "base_image_seq_len": 256,
-  "base_shift": 0.5,
-  "invert_sigmas": false,
-  "max_image_seq_len": 4096,
-  "max_shift": 1.15,
+  "algorithm_type": "dpmsolver++",
+  "flow_shift": 15.0,
   "num_train_timesteps": 1000,
-  "shift": 15.0,
-  "shift_terminal": null,
-  "stochastic_sampling": false,
-  "time_shift_type": "exponential",
-  "use_beta_sigmas": false,
-  "use_dynamic_shifting": false,
-  "use_exponential_sigmas": false,
-  "use_karras_sigmas": false
-}
+  "prediction_type": "flow_prediction",
+  "solver_order": 2,
+  "use_flow_sigmas": true
+}