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

Update forge_reference.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 19:34:53 -08:00
parent 754eb669d0
commit 2b58f91505
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -70,7 +70,9 @@ class PreprocessorReference(Preprocessor):
return h
if flag != 'after':
return
return h
location = transformer_options['block']
sigma = transformer_options["sigmas"][0].item()
if not (sigma_min <= sigma <= sigma_max):
@@ -91,6 +93,9 @@ class PreprocessorReference(Preprocessor):
if not (sigma_min <= sigma <= sigma_max):
return sdp(q, k, v, transformer_options)
location = (transformer_options['block'][0], transformer_options['block'][1],
transformer_options['block_index'])
if self.is_recording_style:
a = 0
else: