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

Perturbed Attention Guidance Integrated

Browse Source
This commit is contained in:
layerdiffusion
2024-07-26 15:00:31 -07:00
parent 1d185584a7
commit 1673a5ac2d
2 changed files with 70 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
modules_forge/unet_patcher.py
View File

@@ -196,3 +196,16 @@ class UnetPatcher(ModelPatcher):
self.add_patches(patches=patch_flat, strength_patch=float(strength), strength_model=1.0)
return
def copy_and_update_model_options(model_options, patch, name, block_name, number, transformer_index=None):
model_options = model_options.copy()
transformer_options = model_options.get("transformer_options", {}).copy()
patches_replace = transformer_options.get("patches_replace", {}).copy()
name_patches = patches_replace.get(name, {}).copy()
block = (block_name, number, transformer_index) if transformer_index is not None else (block_name, number)
name_patches[block] = patch
patches_replace[name] = name_patches
transformer_options["patches_replace"] = patches_replace
model_options["transformer_options"] = transformer_options
return model_options