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: