Update src/pipelines/instantid_pipeline.py

src/pipelines/instantid_pipeline.py

@@ -763,6 +763,6 @@ def get_face_embedding(face_app, ref_images):
         face_info = face_app.get(cv2.cvtColor(np.array(face_image), cv2.COLOR_RGB2BGR))
         face_info = sorted(face_info, key=lambda x: (x['bbox'][2] - x['bbox'][0]) * x['bbox'][3] - x['bbox'][1])[0]  # only use the maximum face
         face_emb = face_info['embedding']
-        emb_list.append(face_emb.cuda())
+        emb_list.append(face_emb)
         # face_kps = draw_kps(face_image, face_info['kps'])
     return emb_list