update

2025-12-06 20:44:33 +08:00 · 2024-12-19 07:57:19 +01:00
1 changed files with 1 additions and 0 deletions
--- a/src/diffusers/models/transformers/transformer_mochi.py
+++ b/src/diffusers/models/transformers/transformer_mochi.py
@@ -334,6 +334,7 @@ class MochiTransformer3DModel(ModelMixin, ConfigMixin, PeftAdapterMixin):
    """

    _supports_gradient_checkpointing = True
+    _no_split_modules = ["MochiTransformerBlock"]

    @register_to_config
    def __init__(