add stable audio small pingpong sampler support

2025-05-15 06:35:17 +00:00
parent b313d05299
commit 05c58339f0
3 changed files with 353 additions and 4 deletions
@@ -26,6 +26,9 @@ from acestep.schedulers.scheduling_flow_match_euler_discrete import (
 from acestep.schedulers.scheduling_flow_match_heun_discrete import (
    FlowMatchHeunDiscreteScheduler,
 )
+from acestep.schedulers.scheduling_flow_match_pingpong import (
+    FlowMatchPingPongScheduler,
+)
 from diffusers.pipelines.stable_diffusion_3.pipeline_stable_diffusion_3 import (
    retrieve_timesteps,
 )
@@ -1043,6 +1046,11 @@ class ACEStepPipeline:
                num_train_timesteps=1000,
                shift=3.0,
            )
+        elif scheduler_type == "pingpong":
+            scheduler = FlowMatchPingPongScheduler(
+                num_train_timesteps=1000,
+                shift=3.0,
+            )

        frame_length = int(duration * 44100 / 512 / 8)
        if src_latents is not None: