From 0df61b5032e6b4d8225783d9742ed5e1cbbb2f11 Mon Sep 17 00:00:00 2001 From: Jedrzej Kosinski Date: Wed, 1 Oct 2025 21:21:36 -0700 Subject: [PATCH] Fix improper index slicing for flipflop get blocks, add extra log message --- comfy/ldm/flipflop_transformer.py | 2 +- comfy/model_patcher.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/comfy/ldm/flipflop_transformer.py b/comfy/ldm/flipflop_transformer.py index a88613c01..0e980ecda 100644 --- a/comfy/ldm/flipflop_transformer.py +++ b/comfy/ldm/flipflop_transformer.py @@ -33,7 +33,7 @@ class FlipFlopModule(torch.nn.Module): if block_type not in self.block_types: raise ValueError(f"Block type {block_type} not found in {self.block_types}") if block_type in self.flipflop: - return getattr(self, block_type)[:self.flipflop[block_type].flip_amount] + return getattr(self, block_type)[:self.flipflop[block_type].i_offset] return getattr(self, block_type) def get_all_block_module_sizes(self, reverse_sort_by_size: bool = False) -> list[tuple[str, int]]: diff --git a/comfy/model_patcher.py b/comfy/model_patcher.py index 67d1cb233..f47e5c157 100644 --- a/comfy/model_patcher.py +++ b/comfy/model_patcher.py @@ -619,6 +619,7 @@ class ModelPatcher: def setup_flipflop(self, flipflop_blocks_per_type: dict[str, tuple[int, int]]): if not self.supports_flipflop(): return + logging.info(f"setting up flipflop with {flipflop_blocks_per_type}") self.model.diffusion_model.setup_flipflop_holders(flipflop_blocks_per_type, self.load_device, self.offload_device) def init_flipflop_block_copies(self):