Update app.py

app.py

@@ -20,6 +20,8 @@ from diffusers import (
     StableDiffusionPipeline, ControlNetModel,
     DPMSolverMultistepScheduler, AutoencoderKL,
 )
+# +++ 修正点A: compel ライブラリをインポート +++
+from compel import Compel
 from insightface.app import FaceAnalysis
 
 ##############################################################################
@@ -161,6 +163,13 @@ for emb in EMB_DIR.glob("*.*"):
     except Exception:
         print("emb skip →", emb.name)
 pipe.to(device)
+
+# +++ 修正点B: compel プロセッサを初期化 +++
+compel_proc = Compel(
+    tokenizer=pipe.tokenizer,
+    text_encoder=pipe.text_encoder,
+    truncate_long_prompts=False  # 長いプロンプトを切り捨てない
+)
 print("pipeline ready ✔")
 
 ##############################################################################
@@ -183,10 +192,11 @@ except Exception as e:
 ##############################################################################
 # 4. プロンプト & 生成関数
 ##############################################################################
+# +++ 修正点C: 不要な <lora:...> 記述を削除 +++
 BASE_PROMPT = (
     "masterpiece, ultra-realistic photo of {subject}, "
     "cinematic lighting, shallow depth of field, textured skin, "
-    "Canon EOS R5 85 mm f/1.4, "
+    "Canon EOS R5 85 mm f/1.4"
 )
 NEG_PROMPT = (
     "ng_deepnegative_v1_75t, CyberRealistic_Negative-neg, UnrealisticDream, "
@@ -197,6 +207,7 @@ 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,
@@ -213,9 +224,13 @@ 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)
+
     result = pipe(
-        prompt=prompt,
-        negative_prompt=neg,
+        prompt_embeds=prompt_embeds,
+        negative_prompt_embeds=negative_prompt_embeds,
         ip_adapter_image=img_in,
         #image=img_in,
         #controlnet_conditioning_scale=0.9,