From 90f49d6665cd1f437bc05ba47d669dbf6bd58209 Mon Sep 17 00:00:00 2001 From: smart8986 Date: Thu, 2 Jul 2026 16:28:24 +0800 Subject: [PATCH 1/2] Fix typo in decorator for get_current_model_index --- lightx2v/models/runners/wan/wan_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightx2v/models/runners/wan/wan_runner.py b/lightx2v/models/runners/wan/wan_runner.py index 8d6b918c8..32d297593 100755 --- a/lightx2v/models/runners/wan/wan_runner.py +++ b/lightx2v/models/runners/wan/wan_runner.py @@ -691,7 +691,7 @@ def infer(self, inputs): self.model[1] = low_noise_model self.model[1].infer(inputs) - @ProfilingContext4DebugL2("Swtich models in infer_main costs") + @ProfilingContext4DebugL2("Switch models in infer_main costs") def get_current_model_index(self): if self.scheduler.timesteps[self.scheduler.step_index] >= self.boundary_timestep: logger.info(f"using - HIGH - noise model at step_index {self.scheduler.step_index + 1}") From d7b4359e85ac9377b98c80fe72a6536336a519fa Mon Sep 17 00:00:00 2001 From: smart8986 Date: Thu, 2 Jul 2026 16:28:43 +0800 Subject: [PATCH 2/2] Fix typo in profiling decorator for model switching --- lightx2v/models/runners/wan/wan_distill_runner.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lightx2v/models/runners/wan/wan_distill_runner.py b/lightx2v/models/runners/wan/wan_distill_runner.py index a981ea6a2..f2d6e0d62 100755 --- a/lightx2v/models/runners/wan/wan_distill_runner.py +++ b/lightx2v/models/runners/wan/wan_distill_runner.py @@ -52,7 +52,7 @@ def __init__(self, model_list, config, boundary_step_index=2): self.cur_model_index = -1 logger.info(f"boundary step index: {self.boundary_step_index}") - @ProfilingContext4DebugL2("Swtich models in infer_main costs") + @ProfilingContext4DebugL2("Switch models in infer_main costs") def get_current_model_index(self): if self.scheduler.step_index < self.boundary_step_index: logger.info(f"using - HIGH - noise model at step_index {self.scheduler.step_index + 1}")