[scheduler] Add SDE Samplers: Curved Euler and Overshoot

docs/source/en/_toctree.yml

@@ -688,6 +688,10 @@
       title: FlowMatchEulerDiscreteScheduler
     - local: api/schedulers/flow_match_heun_discrete
       title: FlowMatchHeunDiscreteScheduler
+    - local: api/schedulers/flow_match_stochastic_curved_euler
+      title: FlowMatchStochasticCurvedEuler
+    - local: api/schedulers/flow_match_stochastic_overshoot
+      title: FlowMatchStochasticOvershoot
     - local: api/schedulers/heun
       title: HeunDiscreteScheduler
     - local: api/schedulers/ipndm

docs/source/en/api/schedulers/flow_match_stochastic_curved_euler.md

@@ -0,0 +1,18 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# FlowMatchStochasticCurvedEulerScheduler
+
+`FlowMatchStochasticCurvedEulerScheduler` is based on the stochastic flow-matching sampling variant introduced in [AMO Sampler](https://huggingface.co/papers/2506.15864).
+
+## FlowMatchStochasticCurvedEulerScheduler
+[[autodoc]] FlowMatchStochasticCurvedEulerScheduler

docs/source/en/api/schedulers/flow_match_stochastic_overshoot.md

@@ -0,0 +1,18 @@
+<!--Copyright 2025 The HuggingFace Team. All rights reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
+the License. You may obtain a copy of the License at
+
+http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
+specific language governing permissions and limitations under the License.
+-->
+
+# FlowMatchStochasticOvershootScheduler
+
+`FlowMatchStochasticOvershootScheduler` is based on the stochastic flow-matching sampling variant introduced in [AMO Sampler](https://huggingface.co/papers/2411.19415).
+
+## FlowMatchStochasticOvershootScheduler
+[[autodoc]] FlowMatchStochasticOvershootScheduler

src/diffusers/__init__.py

@@ -308,6 +308,8 @@
             "FlowMatchEulerDiscreteScheduler",
             "FlowMatchHeunDiscreteScheduler",
             "FlowMatchLCMScheduler",
+            "FlowMatchStochasticCurvedEulerScheduler",
+            "FlowMatchStochasticOvershootDiscreteScheduler",
             "HeunDiscreteScheduler",
             "IPNDMScheduler",
             "KarrasVeScheduler",
@@ -964,6 +966,8 @@
             FlowMatchEulerDiscreteScheduler,
             FlowMatchHeunDiscreteScheduler,
             FlowMatchLCMScheduler,
+            FlowMatchStochasticCurvedEulerScheduler,
+            FlowMatchStochasticOvershootDiscreteScheduler,
             HeunDiscreteScheduler,
             IPNDMScheduler,
             KarrasVeScheduler,

src/diffusers/schedulers/__init__.py

@@ -61,6 +61,8 @@
     _import_structure["scheduling_flow_match_euler_discrete"] = ["FlowMatchEulerDiscreteScheduler"]
     _import_structure["scheduling_flow_match_heun_discrete"] = ["FlowMatchHeunDiscreteScheduler"]
     _import_structure["scheduling_flow_match_lcm"] = ["FlowMatchLCMScheduler"]
+    _import_structure["scheduling_flow_match_stochastic_curved_euler"] = ["FlowMatchStochasticCurvedEulerScheduler"]
+    _import_structure["scheduling_flow_match_stochastic_overshoot"] = ["FlowMatchStochasticOvershootDiscreteScheduler"]
     _import_structure["scheduling_heun_discrete"] = ["HeunDiscreteScheduler"]
     _import_structure["scheduling_ipndm"] = ["IPNDMScheduler"]
     _import_structure["scheduling_k_dpm_2_ancestral_discrete"] = ["KDPM2AncestralDiscreteScheduler"]
@@ -163,6 +165,8 @@
         from .scheduling_flow_match_euler_discrete import FlowMatchEulerDiscreteScheduler
         from .scheduling_flow_match_heun_discrete import FlowMatchHeunDiscreteScheduler
         from .scheduling_flow_match_lcm import FlowMatchLCMScheduler
+        from .scheduling_flow_match_stochastic_curved_euler import FlowMatchStochasticCurvedEulerScheduler
+        from .scheduling_flow_match_stochastic_overshoot import FlowMatchStochasticOvershootDiscreteScheduler
         from .scheduling_heun_discrete import HeunDiscreteScheduler
         from .scheduling_ipndm import IPNDMScheduler
         from .scheduling_k_dpm_2_ancestral_discrete import KDPM2AncestralDiscreteScheduler