added. check for auraflow and pipleline custom

check_app.py

@@ -175,12 +175,23 @@ def create_pipeline_logic(prompt_text, model_name):
     seed = 42
     config = MODEL_CONFIGS[model_name]
     pipe_class = config["pipeline_class"]
-    pipe = AutoPipelineForText2Image.from_pretrained(
-        config["repo_id"],
-        variant="fp16",
-        #cache_dir=config["cache_dir"],
-        torch_dtype=torch.bfloat16
-    ).to("cuda")
+    pipe = None
+    if model_name == "AuraFlow":
+        print("AuraFlow specal")
+        b_pipe = AutoPipelineForText2Image.from_pretrained(
+            config["repo_id"],
+            variant="fp16",
+            #cache_dir=config["cache_dir"],
+            torch_dtype=torch.bfloat16
+        ).to("cuda")
+        pipe = ProgressAuraFlowPipeline(b_pipe)
+    else:
+        pipe = AutoPipelineForText2Image.from_pretrained(
+            config["repo_id"],
+            variant="fp16",
+            #cache_dir=config["cache_dir"],
+            torch_dtype=torch.bfloat16
+        ).to("cuda")
         
     image = generate_image_with_progress(
         pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, seed=seed, progress=progress