diffsynth-engine 0.6.1.dev15__tar.gz → 0.6.1.dev17__tar.gz

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.6.1.dev15
+Version: 0.6.1.dev17
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine/models/flux/flux_dit.py

@@ -515,5 +515,8 @@ class FluxDiT(PreTrainedModel):
         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-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine/models/wan/wan_dit.py

@@ -499,8 +499,5 @@ class WanDiT(PreTrainedModel):
         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}

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine/pipelines/flux_image.py

@@ -573,7 +573,7 @@ class FluxImagePipeline(BasePipeline):
         self.update_component(self.vae_encoder, state_dicts.vae, self.config.device, self.config.vae_dtype)
 
     def compile(self):
-        self.dit.compile_repeated_blocks(dynamic=True)
+        self.dit.compile_repeated_blocks()
 
     def load_loras(self, lora_list: List[Tuple[str, float]], fused: bool = True, save_original_weight: bool = False):
         assert self.config.tp_degree is None or self.config.tp_degree == 1, (

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine/pipelines/qwen_image.py

@@ -307,7 +307,7 @@ class QwenImagePipeline(BasePipeline):
         self.update_component(self.vae, state_dicts.vae, self.config.device, self.config.vae_dtype)
 
     def compile(self):
-        self.dit.compile_repeated_blocks(dynamic=True)
+        self.dit.compile_repeated_blocks()
 
     def load_loras(self, lora_list: List[Tuple[str, float]], fused: bool = True, save_original_weight: bool = False):
         assert self.config.tp_degree is None or self.config.tp_degree == 1, (

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine/pipelines/wan_video.py

@@ -645,6 +645,6 @@ class WanVideoPipeline(BasePipeline):
         return vae_type
 
     def compile(self):
-        self.dit.compile_repeated_blocks(dynamic=True)
+        self.dit.compile_repeated_blocks()
         if self.dit2 is not None:
-            self.dit2.compile_repeated_blocks(dynamic=True)
+            self.dit2.compile_repeated_blocks()

{diffsynth_engine-0.6.1.dev15 → diffsynth_engine-0.6.1.dev17}/diffsynth_engine.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.6.1.dev15
+Version: 0.6.1.dev17
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent