From 4ecad10a66ff88d7d8dcf96d711e333f0014977f Mon Sep 17 00:00:00 2001 From: ciaranbor Date: Thu, 11 Dec 2025 16:17:43 +0000 Subject: [PATCH] Fix group.size() --- .../worker/engines/mflux/pipefusion/distributed_denoising.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py b/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py index 149be0a6..917e2894 100644 --- a/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py +++ b/src/exo/worker/engines/mflux/pipefusion/distributed_denoising.py @@ -60,7 +60,7 @@ class DistributedDenoising: self.end_layer = shard_metadata.end_layer self.num_sync_steps = num_sync_steps - self.num_patches = num_patches if num_patches else group.size + self.num_patches = num_patches if num_patches else group.size() # Get block counts from the original transformer (before slicing) # Note: These are the ORIGINAL counts, not the sliced counts