Update app.py

app.py

@@ -20,7 +20,6 @@ from diffusers import (
     StableDiffusionPipeline, ControlNetModel,
     DPMSolverMultistepScheduler, AutoencoderKL,
 )
-# +++ 修正点A: compel ライブラリをインポート +++
 from compel import Compel
 from insightface.app import FaceAnalysis
 
@@ -164,7 +163,7 @@ for emb in EMB_DIR.glob("*.*"):
         print("emb skip →", emb.name)
 pipe.to(device)
 
-# +++ 修正点B: compel プロセッサを初期化 +++
+# compel プロセッサを初期化
 compel_proc = Compel(
     tokenizer=pipe.tokenizer,
     text_encoder=pipe.text_encoder,
@@ -192,7 +191,6 @@ except Exception as e:
 ##############################################################################
 # 4. プロンプト & 生成関数
 ##############################################################################
-# +++ 修正点C: 不要な <lora:...> 記述を削除 +++
 BASE_PROMPT = (
     "masterpiece, ultra-realistic photo of {subject}, "
     "cinematic lighting, shallow depth of field, textured skin, "
@@ -207,7 +205,6 @@ NEG_PROMPT = (
     "missing arms, missing legs, (badhandv4:0.7), BadNegAnatomyV1-neg, skin blemishes, acnes, age spot, glans"
 )
 
-# +++ 修正点D: generate関数全体をcompel対応版に書き換え +++
 @spaces.GPU(duration=90)
 def generate(
     face_np, subject, add_prompt, add_neg, cfg, ip_scale, steps, w, h, upscale, up_factor,
@@ -224,9 +221,8 @@ def generate(
     pipe.set_ip_adapter_scale(ip_scale)
     img_in = Image.fromarray(face_np)
 
-    # compel を使ってプロンプトをエンベディングに変換
-    prompt_embeds = compel_proc(prompt)
-    negative_prompt_embeds = compel_proc(neg)
+    # +++ 最終修正: compelを一度に呼び出し、プロンプトとネガティブプロンプトの長さを揃える +++
+    prompt_embeds, negative_prompt_embeds = compel_proc([prompt, neg])
 
     result = pipe(
         prompt_embeds=prompt_embeds,