Update pipeline.py

pipeline.py

@@ -58,18 +58,24 @@ class IPFacePlusV2Pipeline:
     
         # Convert face embedding to torch tensor and match dtype
         face_embedding = torch.tensor(faces[0].normed_embedding).to(self.device)
-        if self.pipe.dtype == torch.float16:
-            face_embedding = face_embedding.half()
+    
+        # 安全な dtype チェック（pipe 自体に dtype が無ければ unet から取得）
+        pipe_dtype = getattr(self.pipe, "dtype", getattr(self.pipe, "unet", None).dtype)
+    
+        if pipe_dtype == torch.float16:
+            face_embedding = face_embedding.to(dtype=torch.float16)
+        else:
+            face_embedding = face_embedding.to(dtype=torch.float32)
     
         # Generate image
         image = self.ip_adapter.generate(
             prompt=prompt,
             scale=scale,
             faceid_embeds=face_embedding.unsqueeze(0)  # バッチ次元を追加
-        )[0]  # 最初の1枚を取得（必要に応じて変更）
+        )[0]  # 最初の1枚を取得
     
         # Convert PIL image to base64
         buffered = io.BytesIO()
         image.save(buffered, format="PNG")
         encoded_img = base64.b64encode(buffered.getvalue()).decode("utf-8")
-        return f"data:image/png;base64,{encoded_img}"
+        return f"data:image/png;base64,{encoded_img}"