From ac3f2859931513181f1c3ceaaacd8bce6c4f4889 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Thu, 27 Mar 2025 11:17:14 +0200 Subject: [PATCH] Update nodes.py --- nodes.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/nodes.py b/nodes.py index ecc37e0..08f9ec5 100644 --- a/nodes.py +++ b/nodes.py @@ -2062,20 +2062,21 @@ class WanVideoSampler: else: image_cond_input = torch.cat([torch.zeros_like(image_cond), image_cond]) - if not control_start_percent <= current_step_percentage <= control_end_percent: - if control_lora: - image_cond_input = None + if control_lora: + image_cond_input = None + if not control_start_percent <= current_step_percentage <= control_end_percent: control_lora_enabled = False if patcher.model.is_patched: log.info("Unloading LoRA...") patcher.unpatch_model(device) patcher.model.is_patched = False - else: - if control_lora: + else: if not patcher.model.is_patched: log.info("Loading LoRA...") patcher = apply_lora(patcher, device, device, low_mem_load=False) patcher.model.is_patched = True + else: + image_cond_input = image_cond base_params = { 'seq_len': seq_len,