Update app.py

app.py

@@ -11,6 +11,7 @@ from diffusers import AutoencoderKL, DiffusionPipeline, DPMSolverMultistepSchedu
 from PIL import Image
 from lpw_stable_diffusion_xl import SDXLLongPromptWeightingPipeline
 from compel import Compel, ReturnedEmbeddingsType
+from compel.multi_model_wrappers import CompelForSDXL
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "2048"))
@@ -30,12 +31,21 @@ pipe = SDXLLongPromptWeightingPipeline.from_pretrained(
     trust_remote_code=True,
 ).to(device)
 
-compel = Compel(
-    tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
-    text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
-    # returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES,
+# compel = Compel(
+#     tokenizer=[pipe.tokenizer, pipe.tokenizer_2],
+#     text_encoder=[pipe.text_encoder, pipe.text_encoder_2],
+#     # returned_embeddings_type=ReturnedEmbeddingsType.PENULTIMATE_HIDDEN_STATES,
+# )
+
+
+compel = CompelForSDXL(
+    tokenizer=pipe.tokenizer,
+    tokenizer_2=pipe.tokenizer_2,
+    text_encoder=pipe.text_encoder,
+    text_encoder_2=pipe.text_encoder_2,
 )
 
+
 pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
 pipe.scheduler.config["use_karras_sigmas"] = True
 pipe.scheduler.config["algorithm_type"] = "sde-dpmsolver++"