From df70ff5f212009bd5eea45ce785daf1d9d70cd96 Mon Sep 17 00:00:00 2001 From: lllyasviel Date: Tue, 30 Jan 2024 11:01:49 -0800 Subject: [PATCH] Update preprocessor_revision.py --- .../scripts/preprocessor_revision.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/extensions-builtin/forge_preprocessor_revision/scripts/preprocessor_revision.py b/extensions-builtin/forge_preprocessor_revision/scripts/preprocessor_revision.py index eae93fce..494b0d76 100644 --- a/extensions-builtin/forge_preprocessor_revision/scripts/preprocessor_revision.py +++ b/extensions-builtin/forge_preprocessor_revision/scripts/preprocessor_revision.py @@ -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