From ec035e30f0dbaaf873de398f1afd716a85a65b74 Mon Sep 17 00:00:00 2001 From: travix Date: Wed, 24 Jul 2024 11:06:38 +0800 Subject: [PATCH] Fix: Define zero_locator_tensor before use to prevent undefined variable error --- src/pipelines/pipeline_echo_mimic_pose.py | 1 + 1 file changed, 1 insertion(+) 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)