Update app.py

app.py

@@ -15,8 +15,6 @@ import spaces
 
 model_id = "aipicasso/emix-0-4-turbo"
 auth_token=os.environ["ACCESS_TOKEN"]
-#adapter_id = "latent-consistency/lcm-lora-sdxl"
-#adapter_id_2 = "manual.safetensors"
 
 scheduler = EulerAncestralDiscreteScheduler.from_pretrained(model_id, subfolder="scheduler",token=auth_token)
 
@@ -25,23 +23,9 @@ pipe = AutoPipelineForText2Image.from_pretrained(
   torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32,
   scheduler=scheduler,
   token=auth_token)
-pipe=pipe.to("cuda")
-
-#pipe = AutoPipelineForText2Image.from_pretrained(
-#    model_id, 
-#    torch_dtype=torch.float16,
-#    use_auth_token=auth_token
-#)
-#pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
-
-
-#pipe.load_lora_weights(adapter_id)
-#pipe.load_lora_weights(adapter_id_2)
-#pipe.fuse_lora()
 
+pipe.to("cuda")
 pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.3, b2=1.4)
-#pipe.enable_freeu(s1=0.6, s2=0.4, b1=1.1, b2=1.2)
-#pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
 
 state_dict = load_file("unaestheticXLv31.safetensors")
 pipe.load_textual_inversion(state_dict["clip_g"], token="unaestheticXLv31", text_encoder=pipe.text_encoder_2, tokenizer=pipe.tokenizer_2)
@@ -113,13 +97,15 @@ def auto_prompt_correction(prompt_ui,neg_prompt_ui,disable_auto_prompt_correctio
 
 #
 def txt_to_img(prompt, neg_prompt, guidance, steps, width, height, generator):
-    conditioning, pooled = compel([prompt, neg_prompt])
+    conditioning, pooled = compel(prompt)
+    neg_conditioning, neg_pooled = compel(neg_prompt)
+    
 
     result = pipe(
-        prompt_embeds=conditioning[0:1],
-        pooled_prompt_embeds=pooled[0:1], 
-        negative_prompt_embeds=conditioning[1:2], 
-        negative_pooled_prompt_embeds=pooled[1:2],
+        prompt_embeds=conditioning,
+        pooled_prompt_embeds=pooled, 
+        negative_prompt_embeds=neg_conditioning, 
+        negative_pooled_prompt_embeds=neg_pooled,
         num_inference_steps = int(steps),
         guidance_scale = guidance,
         width = width,