better error for retargeting

2024-07-09 22:06:56 +03:00
parent 92529f7ca8
commit 86e91a6e9d
1 changed files with 4 additions and 1 deletions
@@ -72,7 +72,10 @@ class LivePortraitPipeline(object):
        pbar = comfy.utils.ProgressBar(total_frames)

        if inference_cfg.flag_eye_retargeting or inference_cfg.flag_lip_retargeting:
-            driving_landmark_list = crop_info["driving_landmark_list"]
+            if 'driving_landmark_list' in crop_info.keys():
+                driving_landmark_list = crop_info["driving_landmark_list"]
+            else:
+                raise ValueError("Missing driving_landmark_list in crop_info, get it from opt_driving_images in the Cropper node")

        for i in tqdm(range(total_frames), desc='Animating...', total=total_frames):
            source_frame_rgb = self._get_source_frame(