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

Update preprocessor_revision.py

Browse Source
This commit is contained in:
lllyasviel
2024-01-30 11:01:49 -08:00
parent 8f0cec30eb
commit df70ff5f21
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/forge_preprocessor_revision/scripts/preprocessor_revision.py
View File

@@ -41,14 +41,14 @@ def revision_conditioning_modifier(model, x, timestep, uncond, cond, cond_scale,
uncond = copy.deepcopy(uncond)
for c in cond:
a = 0
c['model_conds']['y'].cond[:, :1280] = new_y.clone()
for c in uncond:
a = 0
c['model_conds']['y'].cond[:, :1280] = torch.zeros_like(new_y)
if ignore_prompt:
for c in cond + uncond:
a = 0
c['model_conds']['c_crossattn'].cond = torch.zeros_like(c['model_conds']['c_crossattn'].cond)
return model, x, timestep, uncond, cond, cond_scale, model_options, seed