fffiloni commited on
Commit
a2395b4
Β·
verified Β·
1 Parent(s): 734e194

Add gradio tqdm tracking

Browse files
Files changed (1) hide show
  1. gradio/vace.py +1 -1
gradio/vace.py CHANGED
@@ -221,7 +221,7 @@ class VACEInference:
221
  preview=True)
222
 
223
 
224
- def generate(self, output_gallery, src_video, src_mask, src_ref_image_1, src_ref_image_2, src_ref_image_3, prompt, negative_prompt, shift_scale, sample_steps, context_scale, guide_scale, infer_seed, output_height, output_width, frame_rate, num_frames):
225
  output_height, output_width, frame_rate, num_frames = int(output_height), int(output_width), int(frame_rate), int(num_frames)
226
  src_ref_images = [x for x in [src_ref_image_1, src_ref_image_2, src_ref_image_3] if
227
  x is not None]
 
221
  preview=True)
222
 
223
 
224
+ def generate(self, output_gallery, src_video, src_mask, src_ref_image_1, src_ref_image_2, src_ref_image_3, prompt, negative_prompt, shift_scale, sample_steps, context_scale, guide_scale, infer_seed, output_height, output_width, frame_rate, num_frames, progress=gr.Progress(track_tqdm=True)):
225
  output_height, output_width, frame_rate, num_frames = int(output_height), int(output_width), int(frame_rate), int(num_frames)
226
  src_ref_images = [x for x in [src_ref_image_1, src_ref_image_2, src_ref_image_3] if
227
  x is not None]