lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-05 04:59:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update unet_patcher.py

Browse Source
This commit is contained in:
lllyasviel
2024-02-03 23:03:21 -08:00
parent dc082be7f9
commit 8be7b347aa
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
modules_forge/unet_patcher.py
View File

@@ -112,13 +112,15 @@ class UnetPatcher(ModelPatcher):
class ExampleScript(scripts.Script):
def process_batch(self, p, *args, **kwargs):
unet = p.sd_model.forge_objects.unet
unet = p.sd_model.forge_objects.unet.clone()
def modifier(x):
return x ** 0.5
unet.add_alphas_cumprod_modifier(modifier)
return
p.sd_model.forge_objects.unet = unet
return
This add_alphas_cumprod_modifier is the only patch option that should be used in process_batch()
All other patch options should be called in process_before_every_sampling()