Refactor get_prompt_attention function to remove unnecessary code

tabs/images/handlers.py

@@ -185,13 +185,13 @@ def get_control_mode(controlnet_config: ControlNetReq):
 #     return has_nsfw_concepts[1]
 
 
-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, 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, device=device)
-        return prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds
+# 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, 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, device=device)
+#         return prompt_embeds, prompt_neg_embeds, pooled_prompt_embeds, negative_pooled_prompt_embeds
 
 
 def cleanup(pipeline, loras = None, embeddings = None):
@@ -211,7 +211,11 @@ def gen_img(request: BaseReq | BaseImg2ImgReq | BaseInpaintReq, progress=gr.Prog
     pipeline = pipeline_args["pipeline"]
     try:
         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)
+        # Get Prompt Embeddings
+        if isinstance(pipeline, flux_pipes):
+            positive_prompt_embeds, positive_prompt_pooled = get_weighted_text_embeddings_flux1(pipeline, request.prompt, device=device)
+        elif isinstance(pipeline, sd_pipes):
+            positive_prompt_embeds, negative_prompt_embeds, positive_prompt_pooled, negative_prompt_pooled = get_weighted_text_embeddings_sdxl(pipeline, request.prompt, request.negative_prompt, device=device)
 
         progress(0.5, "Configuring Pipeline")
         # Common Args