Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import spaces
 import torch
 from diffusers import AutoPipelineForText2Image, AutoencoderKL #,EulerDiscreteScheduler
+from sd_embed.embedding_funcs import get_weighted_text_embeddings_sdxl
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -44,9 +45,24 @@ def infer(
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
+    
+    (
+    prompt_embeds
+    , prompt_neg_embeds
+    , pooled_prompt_embeds
+    , negative_pooled_prompt_embeds
+    ) = get_weighted_text_embeddings_sdxl(
+        pipe
+        , prompt = prompt
+        , negative_prompt = negative_prompt
+    )
     image = pipe(
-        prompt=prompt,
-        negative_prompt=negative_prompt,
+        prompt_embeds                   = prompt_embeds
+        , negative_prompt_embeds        = prompt_neg_embeds
+        , pooled_prompt_embeds          = pooled_prompt_embeds
+        , negative_pooled_prompt_embeds = negative_pooled_prompt_embeds
+        #prompt=prompt,
+        #negative_prompt=negative_prompt,
         width=width,
         height=height,
         guidance_scale=guidance_scale,