提交 54789b4b authored 作者: kijai's avatar kijai

update

上级 50837dd4
...@@ -574,7 +574,7 @@ class MimicMotionPipeline(DiffusionPipeline): ...@@ -574,7 +574,7 @@ class MimicMotionPipeline(DiffusionPipeline):
range(0, num_frames - tile_size + 1, tile_size - tile_overlap)] range(0, num_frames - tile_size + 1, tile_size - tile_overlap)]
if indices[-1][-1] < num_frames - 1: if indices[-1][-1] < num_frames - 1:
indices.append([0, *range(num_frames - tile_size + 1, num_frames)]) indices.append([0, *range(num_frames - tile_size + 1, num_frames)])
comfy_pbar = ProgressBar(len(timesteps)) comfy_pbar = ProgressBar(len(timesteps) * len(indices))
with self.progress_bar(total=len(timesteps) * len(indices)) as progress_bar: with self.progress_bar(total=len(timesteps) * len(indices)) as progress_bar:
for i, t in enumerate(timesteps): for i, t in enumerate(timesteps):
# expand the latents if we are doing classifier free guidance # expand the latents if we are doing classifier free guidance
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论