Update src/diffusers/models/modeling_utils.py

a-r-r-o-w · DN6 · web-flow · commit 2aec312fdbcc · 2025-09-03T20:20:00.000+05:30
Co-authored-by: Dhruv Nair &lt;dhruv.nair@gmail.com&gt;
diff --git a/src/diffusers/models/modeling_utils.py b/src/diffusers/models/modeling_utils.py
@@ -1481,7 +1481,7 @@ def compile_repeated_blocks(self, *args, **kwargs):
                 f"Regional compilation failed because {repeated_blocks} classes are not found in the model. "
             )
 
-    def parallelize(self, *, config: ParallelConfig, cp_plan: Optional[Dict[str, ContextParallelModelPlan]] = None):
+    def enable_parallelism(self, *, config: ParallelConfig, cp_plan: Optional[Dict[str, ContextParallelModelPlan]] = None):
         from ..hooks.context_parallel import apply_context_parallel
 
         logger.warning(

Original file line number	Diff line number	Diff line change
`@@ -1481,7 +1481,7 @@ def compile_repeated_blocks(self, args, *kwargs):`
`1481`	`1481`	`f"Regional compilation failed because {repeated_blocks} classes are not found in the model. "`
`1482`	`1482`	`)`
`1483`	`1483`
`1484`		`- def parallelize(self, *, config: ParallelConfig, cp_plan: Optional[Dict[str, ContextParallelModelPlan]] = None):`
	`1484`	`+ def enable_parallelism(self, *, config: ParallelConfig, cp_plan: Optional[Dict[str, ContextParallelModelPlan]] = None):`
`1485`	`1485`	`from ..hooks.context_parallel import apply_context_parallel`
`1486`	`1486`
`1487`	`1487`	`logger.warning(`