revi13 commited on
Commit
1f8023d
·
verified ·
1 Parent(s): 3f1043d

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +3 -1
pipeline.py CHANGED
@@ -80,7 +80,9 @@ class IPFacePlusV2Pipeline:
80
  image = self.ip_adapter.generate(
81
  prompt=prompt,
82
  scale=scale,
83
- faceid_embeds=face_embedding.unsqueeze(0) # バッチ次元を追加
 
 
84
  )[0] # 最初の1枚を取得
85
 
86
  print("🎨 [pipeline] Convert PIL image to base64")
 
80
  image = self.ip_adapter.generate(
81
  prompt=prompt,
82
  scale=scale,
83
+ faceid_embeds=face_embedding.unsqueeze(0), # バッチ次元を追加
84
+ height=512,
85
+ width=512
86
  )[0] # 最初の1枚を取得
87
 
88
  print("🎨 [pipeline] Convert PIL image to base64")