Update live_portrait_pipeline.py

2024-07-24 21:22:19 +03:00
parent 3336dad236
commit 8b28441984
1 changed files with 2 additions and 0 deletions
@@ -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))