From 613db9aaa2c395787ec3744625a9d71d28c7a3f0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=D0=95=D0=B2=D0=B3=D0=B5=D0=BD=D0=B8=D0=B9=20=D0=93=D1=83?= =?UTF-8?q?=D1=80=D1=8C=D0=B5=D0=B2=20=7C=20Eugene=20Gourieff=20=7C=20?= =?UTF-8?q?=E5=8F=A4=E4=BB=81?= <85128026+Gourieff@users.noreply.github.com> Date: Fri, 19 Jul 2024 23:16:04 +0700 Subject: [PATCH] FIX: name 'zero_locator_tensor' is not defined Issue: https://github.com/BadToBest/EchoMimic/issues/61 --- src/pipelines/pipeline_echo_mimic_pose.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pipelines/pipeline_echo_mimic_pose.py b/src/pipelines/pipeline_echo_mimic_pose.py index fedfa55..6fcbb6c 100644 --- a/src/pipelines/pipeline_echo_mimic_pose.py +++ b/src/pipelines/pipeline_echo_mimic_pose.py @@ -540,8 +540,8 @@ def __call__( audio_latents_cond = torch.cat([audio_fea_final[:, c] for c in new_context]).to(device) audio_latents = torch.cat([torch.zeros_like(audio_latents_cond), audio_latents_cond], 0) pose_latents_cond = torch.cat([face_locator_tensor[:, :, c] for c in new_context]).to(device) - zero_pose_latents = torch.cat([zero_locator_tensor[:, :, c] for c in new_context]).to(device) - pose_latents = torch.cat([torch.zeros_like(zero_pose_latents), pose_latents_cond], 0) + # zero_pose_latents = torch.cat([zero_locator_tensor[:, :, c] for c in new_context]).to(device) + pose_latents = torch.cat([torch.zeros_like(pose_latents_cond), pose_latents_cond], 0) latent_model_input = self.scheduler.scale_model_input( latent_model_input, t