Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
2351462
1
Parent(s):
2c49741
fix pbar for demo 1
Browse files
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 = (
|