small bug fix if multiple kps images are provided

2024-02-23 10:54:59 +01:00
parent 54729bf24c
commit f9c20ad4fa
1 changed files with 1 additions and 1 deletions
@@ -428,7 +428,7 @@ class ApplyInstantID:
        if face_embed is None:
            raise Exception('Reference Image: No face detected.')
        
-        face_kps = extractFeatures(insightface, image_kps if image_kps is not None else image, extract_kps=True)
+        face_kps = extractFeatures(insightface, image_kps[0].unsqueeze(0) if image_kps is not None else image[0].unsqueeze(0), extract_kps=True)

        if face_kps is None:
            face_kps = torch.zeros_like(image) if image_kps is None else image_kps