Spaces:
Build error
Build error
Update app.py
Browse files
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=
|
| 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 |
-
#
|
| 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,
|