lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-29 10:41:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update forge_freeu.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-25 16:11:34 -08:00
parent af82470d78
commit abf33dbb43
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py
View File

@@ -16,7 +16,7 @@ def Fourier_filter(x, threshold, scale):
return x_filtered.to(x.dtype) return x_filtered.to(x.dtype)
def freeu_v2(model, b1, b2, s1, s2): def set_freeu_v2_patch(model, b1, b2, s1, s2):
model_channels = model.model.model_config.unet_config["model_channels"] model_channels = model.model.model_config.unet_config["model_channels"]
scale_dict = {model_channels * 4: (b1, s1), model_channels * 2: (b2, s2)} scale_dict = {model_channels * 4: (b1, s1), model_channels * 2: (b2, s2)}
@@ -64,7 +64,7 @@ class FreeUForForge(scripts.Script):
unet = p.sd_model.forge_objects.unet unet = p.sd_model.forge_objects.unet
unet = freeu_v2(unet, freeu_b1, freeu_b2, freeu_s1, freeu_s2) unet = set_freeu_v2_patch(unet, freeu_b1, freeu_b2, freeu_s1, freeu_s2)
p.sd_model.forge_objects.unet = unet p.sd_model.forge_objects.unet = unet