fffiloni commited on
Commit
d1cceb9
·
verified ·
1 Parent(s): 317c75b

track and log execution time

Browse files
Files changed (1) hide show
  1. app_df.py +7 -0
app_df.py CHANGED
@@ -43,6 +43,8 @@ def generate_diffusion_forced_video(
43
  teacache_thresh=0.2,
44
  use_ret_steps=True,
45
  ):
 
 
46
  model_id = download_model(model_id)
47
 
48
  if resolution == "540P":
@@ -129,6 +131,11 @@ def generate_diffusion_forced_video(
129
  timestamp = time.strftime("%Y%m%d_%H%M%S")
130
  output_path = f"gradio_df_videos/{prompt[:50].replace('/', '')}_{seed}_{timestamp}.mp4"
131
  imageio.mimwrite(output_path, video_frames, fps=fps, quality=8, output_params=["-loglevel", "error"])
 
 
 
 
 
132
  return output_path
133
 
134
 
 
43
  teacache_thresh=0.2,
44
  use_ret_steps=True,
45
  ):
46
+
47
+ start_time = time.perf_counter()
48
  model_id = download_model(model_id)
49
 
50
  if resolution == "540P":
 
131
  timestamp = time.strftime("%Y%m%d_%H%M%S")
132
  output_path = f"gradio_df_videos/{prompt[:50].replace('/', '')}_{seed}_{timestamp}.mp4"
133
  imageio.mimwrite(output_path, video_frames, fps=fps, quality=8, output_params=["-loglevel", "error"])
134
+
135
+ end_time = time.perf_counter()
136
+ elapsed = end_time - start_time
137
+ print(f"[LOG] Execution time: {elapsed:.4f} seconds")
138
+
139
  return output_path
140
 
141