lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-01-31 13:29:46 +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:25:27 -08:00
parent 50c82c06f2
commit 754eb669d0
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -25,6 +25,8 @@ class PreprocessorReference(Preprocessor):
self.do_not_need_model = True
self.is_recording_style = False
self.recorded_attn1 = {}
self.recorded_h = {}
def process_before_every_sampling(self, process, cond, *args, **kwargs):
unit = kwargs['unit']
@@ -46,6 +48,9 @@ class PreprocessorReference(Preprocessor):
sigma_max = unet.model.model_sampling.percent_to_sigma(start_percent)
sigma_min = unet.model.model_sampling.percent_to_sigma(end_percent)
self.recorded_attn1 = {}
self.recorded_h = {}
def conditioning_modifier(model, x, timestep, uncond, cond, cond_scale, model_options, seed):
sigma = timestep[0].item()
if not (sigma_min <= sigma <= sigma_max):