From 245e44b492fee4d711b4c523775b4aa1f81122b5 Mon Sep 17 00:00:00 2001 From: faii <1320407956@qq.com> Date: Wed, 6 Nov 2024 21:23:57 +0800 Subject: [PATCH 1/2] fix repeat problem --- src/pipelines/pipeline_pose2vid_long.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/src/pipelines/pipeline_pose2vid_long.py b/src/pipelines/pipeline_pose2vid_long.py index 1a79410..7b0ad0e 100644 --- a/src/pipelines/pipeline_pose2vid_long.py +++ b/src/pipelines/pipeline_pose2vid_long.py @@ -487,18 +487,6 @@ def __call__( ) num_context_batches = math.ceil(len(context_queue) / context_batch_size) - context_queue = list( - context_scheduler( - 0, - num_inference_steps, - latents.shape[2], - context_frames, - context_stride, - context_overlap, - ) - ) - - num_context_batches = math.ceil(len(context_queue) / context_batch_size) global_context = [] for i in range(num_context_batches): global_context.append( From 4b7d4bda7a23781fcc6b09f6b22889c7c7827104 Mon Sep 17 00:00:00 2001 From: faii <1320407956@qq.com> Date: Wed, 6 Nov 2024 21:27:53 +0800 Subject: [PATCH 2/2] fix repeat problem --- src/pipelines/pipeline_pose2vid_long.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/pipelines/pipeline_pose2vid_long.py b/src/pipelines/pipeline_pose2vid_long.py index 7b0ad0e..7323a77 100644 --- a/src/pipelines/pipeline_pose2vid_long.py +++ b/src/pipelines/pipeline_pose2vid_long.py @@ -482,9 +482,10 @@ def __call__( latents.shape[2], context_frames, context_stride, - 0, + context_overlap, ) ) + num_context_batches = math.ceil(len(context_queue) / context_batch_size) global_context = []