Update app.py

app.py

@@ -4,7 +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
+from compel import Compel, ReturnedEmbeddingsType
 
 if not torch.cuda.is_available():
     DESCRIPTION += "\n<p>你现在运行在CPU上 但是只支持GPU.</p>"
@@ -45,15 +45,13 @@ def infer(
 ):
     seed = int(randomize_seed_fn(seed, randomize_seed))
     generator = torch.Generator().manual_seed(seed)
+    compel = Compel(tokenizer=pipeline.tokenizer, text_encoder=pipeline.text_encoder)
+    conditioning = compel.build_conditioning_tensor(prompt)
     
-    (prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds) = get_weighted_text_embeddings_sdxl(pipe, prompt = prompt, neg_prompt = negative_prompt)
     image = pipe(
-        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,
+        prompt_embeds=conditioning,
+        negative_prompt=negative_prompt,
         width=width,
         height=height,
         guidance_scale=guidance_scale,