Spaces:

Fucius
/

OMG-InstantID

Running on Zero

Fucius commited on Mar 22

Commit

c836a64

•

1 Parent(s): 91c687c

Update src/pipelines/instantid_pipeline.py

Files changed (1) hide show

src/pipelines/instantid_pipeline.py CHANGED Viewed

@@ -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,
                                                              concept_models._execution_device,
                                                              num_images_per_prompt,
                                                              concept_models.unet.dtype,

             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(),
                                                              concept_models._execution_device,
                                                              num_images_per_prompt,
                                                              concept_models.unet.dtype,