i0switch commited on
Commit
4d7c512
·
verified ·
1 Parent(s): 7bd2e19

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -205,7 +205,7 @@ NEG_PROMPT = (
205
  "missing arms, missing legs, (badhandv4:0.7), BadNegAnatomyV1-neg, skin blemishes, acnes, age spot, glans"
206
  )
207
 
208
- @spaces.GPU(duration=90)
209
  def generate(
210
  face_np, subject, add_prompt, add_neg, cfg, ip_scale, steps, w, h, upscale, up_factor,
211
  progress=gr.Progress(track_tqdm=True),
@@ -221,8 +221,10 @@ def generate(
221
  pipe.set_ip_adapter_scale(ip_scale)
222
  img_in = Image.fromarray(face_np)
223
 
224
- # +++ 最終修正: compelを一度に呼び出し、プロンプトとネガティブプロンプトの長さを揃える +++
225
  prompt_embeds, negative_prompt_embeds = compel_proc([prompt, neg])
 
 
226
 
227
  result = pipe(
228
  prompt_embeds=prompt_embeds,
 
205
  "missing arms, missing legs, (badhandv4:0.7), BadNegAnatomyV1-neg, skin blemishes, acnes, age spot, glans"
206
  )
207
 
208
+ @spaces.GPU(duration=60)
209
  def generate(
210
  face_np, subject, add_prompt, add_neg, cfg, ip_scale, steps, w, h, upscale, up_factor,
211
  progress=gr.Progress(track_tqdm=True),
 
221
  pipe.set_ip_adapter_scale(ip_scale)
222
  img_in = Image.fromarray(face_np)
223
 
224
+ # compelで長さを揃え、.unsqueeze(0)でバッチ次元を追加する
225
  prompt_embeds, negative_prompt_embeds = compel_proc([prompt, neg])
226
+ prompt_embeds = prompt_embeds.unsqueeze(0)
227
+ negative_prompt_embeds = negative_prompt_embeds.unsqueeze(0)
228
 
229
  result = pipe(
230
  prompt_embeds=prompt_embeds,