From abf33dbb433c657f4a07134eb081cb6f53cf900f Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Thu, 25 Jan 2024 16:11:34 -0800 Subject: [PATCH] Update forge_freeu.py --- extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py b/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py index 756e094d..eb1b7021 100644 --- a/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py +++ b/extensions-builtin/sd_forge_freeu/scripts/forge_freeu.py @@ -16,7 +16,7 @@ def Fourier_filter(x, threshold, scale): 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"] 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 = 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