Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
·
6d37dbb
1
Parent(s):
1ecf693
disable callback
Browse files- 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()
|