Spaces:
Running
on
Zero
Running
on
Zero
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,
|