OpenSourceJoker/LivePortrait
3
0
Fork 0
mirror of https://github.com/KwaiVGI/LivePortrait.git synced 2024-12-22 20:42:38 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

doc: correct retarget_eye output size

Browse Source
This commit is contained in:
guojianzhu 2024-07-14 22:47:33 +08:00
parent b370d8059c
commit 1613690d69
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
src/live_portrait_wrapper.py
View File

@ -52,9 +52,9 @@ class LivePortraitWrapper(object):
# Optimize for inference # Optimize for inference
if self.compile: if self.compile:
torch._dynamo.config.suppress_errors = True # Suppress errors and fall back to eager execution 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.warping_module = torch.compile(self.warping_module, mode='max-autotune')
self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune') self.spade_generator = torch.compile(self.spade_generator, mode='max-autotune')
self.timer = Timer() self.timer = Timer()
def update_config(self, user_args): def update_config(self, user_args):
@ -199,7 +199,7 @@ class LivePortraitWrapper(object):
""" """
kp_source: BxNx3 kp_source: BxNx3
eye_close_ratio: Bx3 eye_close_ratio: Bx3
Return: Bx(3*num_kp+2) Return: Bx(3*num_kp)
""" """
feat_eye = concat_feat(kp_source, eye_close_ratio) feat_eye = concat_feat(kp_source, eye_close_ratio)
@ -212,6 +212,7 @@ class LivePortraitWrapper(object):
""" """
kp_source: BxNx3 kp_source: BxNx3
lip_close_ratio: Bx2 lip_close_ratio: Bx2
Return: Bx(3*num_kp)
""" """
feat_lip = concat_feat(kp_source, lip_close_ratio) feat_lip = concat_feat(kp_source, lip_close_ratio)