From 8b2844198462c052e218233f1b0acd80f1fc72bb Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Wed, 24 Jul 2024 21:22:19 +0300 Subject: [PATCH] Update live_portrait_pipeline.py --- liveportrait/live_portrait_pipeline.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/liveportrait/live_portrait_pipeline.py b/liveportrait/live_portrait_pipeline.py index 7747465..68dccc6 100644 --- a/liveportrait/live_portrait_pipeline.py +++ b/liveportrait/live_portrait_pipeline.py @@ -73,6 +73,8 @@ class LivePortraitPipeline(object): driving_info.append(None) driving_rot_list.append(None) driving_exp_list.append(None) + if i == 0: + raise ValueError("No face detected in FIRST source image") continue x_d_info = self.live_portrait_wrapper.get_kp_info(driving_images[i].unsqueeze(0).to(device))