#

2026-04-15 16:39:42 +00:00 · 2025-04-22 16:15:12 +08:00
parent d33d5eccd1
commit 7f984f4775
1 changed files with 2 additions and 1 deletions
--- a/library/flux_models.py
+++ b/library/flux_models.py
@@ -1004,7 +1004,8 @@ class Flux(nn.Module):
        if self.blocks_to_swap is None or self.blocks_to_swap == 0:
            return
        self.offloader_double.prepare_block_devices_before_forward(self.double_blocks)
-        self.offloader_single.prepare_block_devices_before_forward(self.single_blocks)    
+        self.offloader_single.prepare_block_devices_before_forward(self.single_blocks)
+
    def forward(
        self,
        img: Tensor,