zhiweili commited on
Commit
6d37dbb
·
1 Parent(s): 1ecf693

disable callback

Browse files
Files changed (1) hide show
  1. app_video.py +0 -3
app_video.py CHANGED
@@ -31,8 +31,6 @@ def create_demo() -> gr.Blocks:
31
  progress: gr.Progress = gr.Progress(),
32
  ):
33
  generator = torch.Generator(device=device).manual_seed(seed)
34
- def gradio_progress_callback(self, step, timestep, kwargs):
35
- progress((step + 1) / num_inference_steps)
36
  run_task_time = 0
37
  time_cost_str = ''
38
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
@@ -46,7 +44,6 @@ def create_demo() -> gr.Blocks:
46
  height=height,
47
  num_frames=num_frames,
48
  num_inference_steps=num_inference_steps,
49
- callback_on_step_end=gradio_progress_callback,
50
  ).frames[0]
51
  finally:
52
  torch.cuda.empty_cache()
 
31
  progress: gr.Progress = gr.Progress(),
32
  ):
33
  generator = torch.Generator(device=device).manual_seed(seed)
 
 
34
  run_task_time = 0
35
  time_cost_str = ''
36
  run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
 
44
  height=height,
45
  num_frames=num_frames,
46
  num_inference_steps=num_inference_steps,
 
47
  ).frames[0]
48
  finally:
49
  torch.cuda.empty_cache()