Update pipline_StableDiffusionXL_ConsistentID.py

pipline_StableDiffusionXL_ConsistentID.py

@@ -261,15 +261,16 @@ class ConsistentIDStableDiffusionXLPipeline(StableDiffusionXLPipeline):
                 attn_processor.scale = scale
 
     @torch.inference_mode()
-    def get_prepare_faceid(self, input_image_path=None):
-        faceid_image = cv2.imread(input_image_path)
+    def get_prepare_faceid(self,input_image_file=None, input_image_path=None):
+        # faceid_image = cv2.imread(input_image_path) ### path may error
+        faceid_image = cv2.cvtColor(np.array(input_image_file), cv2.COLOR_RGB2BGR)
         face_info = self.app.get(faceid_image)
         if face_info==[]:
             faceid_embeds = torch.zeros_like(torch.empty((1, 512)))
         else:
             faceid_embeds = torch.from_numpy(face_info[0].normed_embedding).unsqueeze(0)
 
-        # print(f"faceid_embeds is : {faceid_embeds}")
+        print(f" ========== faceid_embeds is : {faceid_embeds} ==========\r\n")
         return faceid_embeds
 
     @torch.inference_mode()
@@ -507,7 +508,7 @@ class ConsistentIDStableDiffusionXLPipeline(StableDiffusionXLPipeline):
         do_classifier_free_guidance = guidance_scale >= 1.0
         input_image_file = input_id_images[0]
 
-        faceid_embeds = self.get_prepare_faceid(input_image_path=input_image_path)
+        faceid_embeds = self.get_prepare_faceid(input_image_file=input_image_file, input_image_path=input_image_path)
         face_caption = self.get_prepare_llva_caption(input_image_file=input_image_file)
         key_parsing_mask_list, vis_parsing_anno_color = self.get_prepare_facemask(input_image_file)