Update src/pipelines/instantid_pipeline.py

src/pipelines/instantid_pipeline.py

@@ -380,7 +380,7 @@ class InstantidMultiConceptPipeline(StableDiffusionXLControlNetPipeline):
             image_embedding_list = get_face_embedding(face_app, ref_images)
             image_prompt_image_emb_list = []
             for image_embeds in image_embedding_list:
-                prompt_image_emb = concept_models._encode_prompt_image_emb(image_embeds.cuda(),
+                prompt_image_emb = concept_models._encode_prompt_image_emb(image_embeds,
                                                              concept_models._execution_device,
                                                              num_images_per_prompt,
                                                              concept_models.unet.dtype,
@@ -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)
+        emb_list.append(face_emb.cuda())
         # face_kps = draw_kps(face_image, face_info['kps'])
     return emb_list