From ee8c09416b73f9fe45921bce71f01a6f120f6666 Mon Sep 17 00:00:00 2001
From: flyingshan <42313652+flyingshan@users.noreply.github.com>
Date: Fri, 1 Aug 2025 13:38:15 +0800
Subject: [PATCH] fix(wan2.2 train): Fix high/low noise timestep schedule.
 (#270)

---
 scripts/wan2.2/train.py      | 4 ++--
 scripts/wan2.2/train_lora.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/scripts/wan2.2/train.py b/scripts/wan2.2/train.py
index 011b4d3..d8acc9d 100644
--- a/scripts/wan2.2/train.py
+++ b/scripts/wan2.2/train.py
@@ -1480,10 +1480,10 @@ def main():
         vae_stream_2 = None
 
     boundary = config['transformer_additional_kwargs'].get('boundary', 0.900)
-    if args.boundary_type == "low":
+    if args.boundary_type == "high":
         start_num_idx = 0
         train_sampling_steps = int(args.train_sampling_steps * boundary)
-    elif args.boundary_type == "high":
+    elif args.boundary_type == "low":
         start_num_idx = int(args.train_sampling_steps * boundary)
         train_sampling_steps = args.train_sampling_steps - int(args.train_sampling_steps * boundary)
     else:
diff --git a/scripts/wan2.2/train_lora.py b/scripts/wan2.2/train_lora.py
index 7235d2f..fb80a61 100755
--- a/scripts/wan2.2/train_lora.py
+++ b/scripts/wan2.2/train_lora.py
@@ -1488,10 +1488,10 @@ def main():
         vae_stream_2 = None
 
     boundary = config['transformer_additional_kwargs'].get('boundary', 0.900)
-    if args.boundary_type == "low":
+    if args.boundary_type == "high":
         start_num_idx = 0
         train_sampling_steps = int(args.train_sampling_steps * boundary)
-    elif args.boundary_type == "high":
+    elif args.boundary_type == "low":
         start_num_idx = int(args.train_sampling_steps * boundary)
         train_sampling_steps = args.train_sampling_steps - int(args.train_sampling_steps * boundary)
     else: