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/models/ltx2_dit.py
+++ b/diffsynth/models/ltx2_dit.py
@@ -1280,6 +1280,7 @@ class LTXModel(torch.nn.Module):
    LTX model transformer implementation.
    This class implements the transformer blocks for the LTX model.
    """
+    _repeated_blocks = ["BasicAVTransformerBlock"]

    def __init__(  # noqa: PLR0913
        self,