From 766a49dfcbaf176061607d36315f21f9f093ccae Mon Sep 17 00:00:00 2001 From: Haoming Date: Tue, 5 May 2026 17:58:30 +0800 Subject: [PATCH] tiling --- modules_forge/utils.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/modules_forge/utils.py b/modules_forge/utils.py index d9da2e51..14c123e6 100644 --- a/modules_forge/utils.py +++ b/modules_forge/utils.py @@ -31,6 +31,10 @@ def apply_circular_forge(model, tiling_enabled=False): for layer in [layer for layer in unet.modules() if isinstance(layer, torch.nn.Conv2d)]: layer.padding_mode = "circular" if tiling_enabled else "zeros" + vae: torch.nn.Module = model.forge_objects.vae.first_stage_model + for layer in [layer for layer in vae.modules() if isinstance(layer, torch.nn.Conv2d)]: + layer.padding_mode = "circular" if tiling_enabled else "zeros" + print(f"Tiling: {tiling_enabled}")