Skip to content

GitLab

Commit 6c2f36de authored by Yang Yong(雍洋)'s avatar Yang Yong(雍洋) Committed by GitHub
Browse files

update profile (#273)

parent 89ce2aa6
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 8 deletions
lightx2v/models/runners/default_runner.py
View file @ 6c2f36de
...@@ -217,14 +217,15 @@ class DefaultRunner(BaseRunner): ...@@ -217,14 +217,15 @@ class DefaultRunner(BaseRunner):
def run_main(self, total_steps=None): def run_main(self, total_steps=None):
self.init_run() self.init_run()
for segment_idx in range(self.video_segment_num): for segment_idx in range(self.video_segment_num):
# 1. default do nothing with ProfilingContext4Debug(f"segment end2end {segment_idx}"):
self.init_run_segment(segment_idx) # 1. default do nothing
# 2. main inference loop self.init_run_segment(segment_idx)
latents, generator = self.run_segment(total_steps=total_steps) # 2. main inference loop
# 3. vae decoder latents, generator = self.run_segment(total_steps=total_steps)
self.gen_video = self.run_vae_decoder(latents) # 3. vae decoder
# 4. default do nothing self.gen_video = self.run_vae_decoder(latents)
self.end_run_segment() # 4. default do nothing
self.end_run_segment()
self.end_run() self.end_run()
@ProfilingContext("Run VAE Decoder") @ProfilingContext("Run VAE Decoder")
......
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