Refactor get_prompt_attention function to include device parameter

tabs/images/handlers.py

@@ -187,16 +187,16 @@ def get_control_mode(controlnet_config: ControlNetReq):
 
 def get_prompt_attention(pipeline, prompt, negative_prompt):
     if isinstance(pipeline, flux_pipes):
-        prompt_embeds, pooled_prompt_embeds = get_weighted_text_embeddings_flux1(pipeline, prompt)
+        prompt_embeds, pooled_prompt_embeds = get_weighted_text_embeddings_flux1(pipeline, prompt, device=device)
         return prompt_embeds, None, pooled_prompt_embeds, None
     elif isinstance(pipeline, sd_pipes):
-        prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds = get_weighted_text_embeddings_sdxl(pipeline, prompt, negative_prompt)
+        prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds = get_weighted_text_embeddings_sdxl(pipeline, prompt, negative_prompt, device=device)
         return prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds
 
 
 def cleanup(pipeline, loras = None, embeddings = None):
     if loras:
-        pipeline.disable_lora()
+        # pipeline.disable_lora()
         pipeline.unload_lora_weights()
     if embeddings:
         pipeline.unload_textual_inversion()
@@ -210,9 +210,10 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq, progress=gr.Prog
     pipeline_args = get_pipe(request)
     pipeline = pipeline_args["pipeline"]
     try:
-        progress(0.5, "Configuring Pipeline")
+        progress(0.3, "Getting Prompt Embeddings")
         positive_prompt_embeds, negative_prompt_embeds, positive_prompt_pooled, negative_prompt_pooled = get_prompt_attention(pipeline, request.prompt, request.negative_prompt)
 
+        progress(0.5, "Configuring Pipeline")
         # Common Args
         args = {
             'prompt_embeds': positive_prompt_embeds,