Skip to content

GitLab

Commit 1defaa8d authored by helloyongyang's avatar helloyongyang
Browse files

update logs

parent 2dc83b97
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
lightx2v/models/runners/default_runner.py
View file @ 1defaa8d
...@@ -113,7 +113,7 @@ class DefaultRunner(BaseRunner): ...@@ -113,7 +113,7 @@ class DefaultRunner(BaseRunner):
with ProfilingContext4Debug("step_pre"): with ProfilingContext4Debug("step_pre"):
self.model.scheduler.step_pre(step_index=step_index) self.model.scheduler.step_pre(step_index=step_index)
with ProfilingContext4Debug("infer_main"): with ProfilingContext4Debug("🚀 infer_main"):
self.model.infer(self.inputs) self.model.infer(self.inputs)
with ProfilingContext4Debug("step_post"): with ProfilingContext4Debug("step_post"):
......
lightx2v/models/runners/wan/wan_audio_runner.py
View file @ 1defaa8d
...@@ -360,7 +360,7 @@ class VideoGenerator: ...@@ -360,7 +360,7 @@ class VideoGenerator:
with ProfilingContext4Debug("step_pre"): with ProfilingContext4Debug("step_pre"):
self.model.scheduler.step_pre(step_index=step_index) self.model.scheduler.step_pre(step_index=step_index)
with ProfilingContext4Debug("infer_main"): with ProfilingContext4Debug("🚀 infer_main"):
self.model.infer(inputs) self.model.infer(inputs)
with ProfilingContext4Debug("step_post"): with ProfilingContext4Debug("step_post"):
......
lightx2v/models/runners/wan/wan_causvid_runner.py
View file @ 1defaa8d
...@@ -89,7 +89,7 @@ class WanCausVidRunner(WanRunner): ...@@ -89,7 +89,7 @@ class WanCausVidRunner(WanRunner):
self.model.scheduler.latents = self.model.scheduler.last_sample self.model.scheduler.latents = self.model.scheduler.last_sample
self.model.scheduler.step_pre(step_index=self.model.scheduler.infer_steps - 1) self.model.scheduler.step_pre(step_index=self.model.scheduler.infer_steps - 1)
with ProfilingContext4Debug("infer_main"): with ProfilingContext4Debug("🚀 infer_main"):
self.model.infer(self.inputs, kv_start, kv_end) self.model.infer(self.inputs, kv_start, kv_end)
kv_start += self.num_frame_per_block * self.frame_seq_length kv_start += self.num_frame_per_block * self.frame_seq_length
...@@ -108,7 +108,7 @@ class WanCausVidRunner(WanRunner): ...@@ -108,7 +108,7 @@ class WanCausVidRunner(WanRunner):
with ProfilingContext4Debug("step_pre"): with ProfilingContext4Debug("step_pre"):
self.model.scheduler.step_pre(step_index=step_index) self.model.scheduler.step_pre(step_index=step_index)
with ProfilingContext4Debug("infer_main"): with ProfilingContext4Debug("🚀 infer_main"):
self.model.infer(self.inputs, kv_start, kv_end) self.model.infer(self.inputs, kv_start, kv_end)
with ProfilingContext4Debug("step_post"): with ProfilingContext4Debug("step_post"):
......
lightx2v/models/runners/wan/wan_skyreels_v2_df_runner.py
View file @ 1defaa8d
...@@ -110,7 +110,7 @@ class WanSkyreelsV2DFRunner(WanRunner): # Diffustion foring for SkyReelsV2 DF I ...@@ -110,7 +110,7 @@ class WanSkyreelsV2DFRunner(WanRunner): # Diffustion foring for SkyReelsV2 DF I
with ProfilingContext4Debug("step_pre"): with ProfilingContext4Debug("step_pre"):
self.model.scheduler.step_pre(step_index=step_index) self.model.scheduler.step_pre(step_index=step_index)
with ProfilingContext4Debug("infer_main"): with ProfilingContext4Debug("🚀 infer_main"):
self.model.infer(self.inputs) self.model.infer(self.inputs)
with ProfilingContext4Debug("step_post"): with ProfilingContext4Debug("step_post"):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment