From 45dca5c41c11d56aec63a2db962529f0f6dbb93b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Wed, 4 Jun 2025 10:02:20 +0300
Subject: [PATCH] Fix ATI start/end percent controls

---
 nodes.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/nodes.py b/nodes.py
index d17fa4a..f9bc82e 100644
--- a/nodes.py
+++ b/nodes.py
@@ -2968,10 +2968,10 @@ class WanVideoSampler:
                                 log.info("Loading LoRA...")
                                 patcher = apply_lora(patcher, device, device, low_mem_load=False)
                                 patcher.model.is_patched = True
-                elif ATI_tracks is not None:
-                    if (ati_start_percent <= current_step_percentage <= ati_end_percent) or \
-                        (ati_end_percent > 0 and idx == 0 and current_step_percentage >= ati_start_percent):
-                        image_cond_input = image_cond_ati.to(z)
+                                
+                elif ATI_tracks is not None and ((ati_start_percent <= current_step_percentage <= ati_end_percent) or 
+                              (ati_end_percent > 0 and idx == 0 and current_step_percentage >= ati_start_percent)):
+                    image_cond_input = image_cond_ati.to(z)
                 else:
                     image_cond_input = image_cond.to(z) if image_cond is not None else None