diff --git a/src/pipelines/pipeline_echo_mimic_pose.py b/src/pipelines/pipeline_echo_mimic_pose.py index fedfa55..8206f51 100644 --- a/src/pipelines/pipeline_echo_mimic_pose.py +++ b/src/pipelines/pipeline_echo_mimic_pose.py @@ -457,6 +457,7 @@ def __call__( ) face_locator_tensor = self.face_locator(face_mask_tensor) + zero_locator_tensor = torch.zeros_like(face_locator_tensor) extra_step_kwargs = self.prepare_extra_step_kwargs(generator, eta)