From 1613690d6903c58df8089a05e52fed0946802886 Mon Sep 17 00:00:00 2001 From: guojianzhu Date: Sun, 14 Jul 2024 22:47:33 +0800 Subject: [PATCH] doc: correct retarget_eye output size --- src/live_portrait_wrapper.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/src/live_portrait_wrapper.py b/src/live_portrait_wrapper.py index d2a678d..c7f65c9 100644 --- a/src/live_portrait_wrapper.py +++ b/src/live_portrait_wrapper.py @@ -52,9 +52,9 @@ class LivePortraitWrapper(object): # Optimize for inference if self.compile: torch._dynamo.config.suppress_errors = True # Suppress errors and fall back to eager execution - self.warping_module = torch.compile(self.warping_module, mode='max-autotune') - self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune') - + self.warping_module = torch.compile(self.warping_module, mode='max-autotune') + self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune') + self.timer = Timer() def update_config(self, user_args): @@ -199,7 +199,7 @@ class LivePortraitWrapper(object): """ kp_source: BxNx3 eye_close_ratio: Bx3 - Return: Bx(3*num_kp+2) + Return: Bx(3*num_kp) """ feat_eye = concat_feat(kp_source, eye_close_ratio) @@ -212,6 +212,7 @@ class LivePortraitWrapper(object): """ kp_source: BxNx3 lip_close_ratio: Bx2 + Return: Bx(3*num_kp) """ feat_lip = concat_feat(kp_source, lip_close_ratio)