Support Torch Compile (#1368)

* support simple compile * add support for compile * minor fix * minor fix * minor fix
2026-03-24 18:28:10 +00:00 · 2026-03-24 11:19:43 +08:00
parent e2a3a987da
commit ae8cb139e8
18 changed files with 59 additions and 2 deletions
--- a/diffsynth/pipelines/mova_audio_video.py
+++ b/diffsynth/pipelines/mova_audio_video.py
@@ -52,6 +52,7 @@ class MovaAudioVideoPipeline(BasePipeline):
            MovaAudioVideoUnit_UnifiedSequenceParallel(),
        ]
        self.model_fn = model_fn_mova_audio_video
+        self.compilable_models = ["video_dit", "video_dit2", "audio_dit"]

    def enable_usp(self):
        from ..utils.xfuser import get_sequence_parallel_world_size, usp_attn_forward