lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-07 00:19:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

ini mask support

Browse Source
This commit is contained in:
lllyasviel
2024-02-01 21:21:41 -08:00
parent 07977a1fec
commit 9f6ee2a688
9 changed files with 109 additions and 236 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
View File

@@ -47,7 +47,7 @@ class PreprocessorReference(Preprocessor):
self.recorded_attn1 = {}
self.recorded_h = {}
def process_before_every_sampling(self, process, cond, *args, **kwargs):
def process_before_every_sampling(self, process, cond, mask, *args, **kwargs):
unit = kwargs['unit']
weight = float(unit.weight)
style_fidelity = float(unit.threshold_a)
@@ -190,7 +190,7 @@ class PreprocessorReference(Preprocessor):
process.sd_model.forge_objects.unet = unet
return
return cond, mask
add_supported_preprocessor(PreprocessorReference(