lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-21 06:48:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

GroupNorm patcher (#593)

Browse Source 
* add gn wrapper and corresponding patcher

* add gn wrapper and corresponding patcher
This commit is contained in:
continue revolution
2024-08-02 03:53:27 +08:00
committed by GitHub
parent 0eea1acc6e
commit 5192e912ab
2 changed files with 34 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/unet_patcher.py
View File

@@ -159,6 +159,10 @@ class UnetPatcher(ModelPatcher):
self.append_transformer_option('controlnet_conditioning_modifiers', modifier, ensure_uniqueness)
return
def set_groupnorm_wrapper(self, wrapper):
self.set_transformer_option('groupnorm_wrapper', wrapper)
return
def set_controlnet_model_function_wrapper(self, wrapper):
self.set_transformer_option('controlnet_model_function_wrapper', wrapper)
return