CodeCow/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-07-21 21:01:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

tiling

Browse Source
This commit is contained in:
Haoming 2026-05-05 17:58:30 +08:00
parent 3a3c9e1848
commit 766a49dfcb
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/utils.py
View File

@ -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}")