kiwhansong commited on
Commit
2351462
·
1 Parent(s): 2c49741

fix pbar for demo 1

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -124,13 +124,19 @@ def any_images_to_short_video(
124
  scene_idx: int,
125
  image_indices: List[int],
126
  guidance_scale: float,
127
- progress=gr.Progress(track_tqdm=True),
128
  ):
129
  video = video_list[scene_idx]
130
  poses = poses_list[scene_idx]
131
  indices = torch.linspace(0, video.size(0) - 1, 8, dtype=torch.long)
132
  xs = video[indices].unsqueeze(0).to("cuda")
133
  conditions = poses[indices].unsqueeze(0).to("cuda")
 
 
 
 
 
 
 
134
  gen_video = dfot._unnormalize_x(
135
  dfot._sample_sequence(
136
  batch_size=1,
@@ -143,6 +149,7 @@ def any_images_to_short_video(
143
  guidance_scale=guidance_scale,
144
  visualize=False,
145
  ),
 
146
  )[0]
147
  )
148
  gen_video = (
 
124
  scene_idx: int,
125
  image_indices: List[int],
126
  guidance_scale: float,
 
127
  ):
128
  video = video_list[scene_idx]
129
  poses = poses_list[scene_idx]
130
  indices = torch.linspace(0, video.size(0) - 1, 8, dtype=torch.long)
131
  xs = video[indices].unsqueeze(0).to("cuda")
132
  conditions = poses[indices].unsqueeze(0).to("cuda")
133
+ pbar = CustomProgressBar(
134
+ gr.Progress(track_tqdm=True).tqdm(
135
+ iterable=None,
136
+ desc="Sampling",
137
+ total=dfot.sampling_timesteps,
138
+ )
139
+ )
140
  gen_video = dfot._unnormalize_x(
141
  dfot._sample_sequence(
142
  batch_size=1,
 
149
  guidance_scale=guidance_scale,
150
  visualize=False,
151
  ),
152
+ pbar=pbar,
153
  )[0]
154
  )
155
  gen_video = (