fix compile repeated blocks (#191)

Author: akaitsuki-ii

diffsynth_engine/models/flux/flux_dit.py

@@ -515,5 +515,8 @@ def compile_repeated_blocks(self, *args, **kwargs):
         for block in self.blocks:
             block.compile(*args, **kwargs)
 
+        for block in self.single_blocks:
+            block.compile(*args, **kwargs)
+
     def get_fsdp_module_cls(self):
         return {FluxDoubleTransformerBlock, FluxSingleTransformerBlock}

diffsynth_engine/models/wan/wan_dit.py

@@ -499,8 +499,5 @@ def compile_repeated_blocks(self, *args, **kwargs):
         for block in self.blocks:
             block.compile(*args, **kwargs)
 
-        for block in self.single_blocks:
-            block.compile(*args, **kwargs)
-
     def get_fsdp_module_cls(self):
         return {DiTBlock}