added common ProgressPipeline

check_app.py

@@ -180,22 +180,20 @@ def create_pipeline_logic(prompt_text, model_name):
     config = MODEL_CONFIGS[model_name]
     pipe_class = config["pipeline_class"]
     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")
+    b_pipe = AutoPipelineForText2Image.from_pretrained(
+        config["repo_id"],
+        #variant="fp16",
+        #cache_dir=config["cache_dir"],
+        torch_dtype=torch.bfloat16
+    ).to("cuda")
+    pipe_signature = signature(b_pipe)
+    # Check for the presence of "callback_on_step_end" in the signature
+    has_callback_on_step_end = "callback_on_step_end" in pipe_signature.parameters
+    if not has_callback_on_step_end
         pipe = ProgressPipeline(b_pipe)
+        print("ProgressPipeline specal")
     else:
-        pipe = AutoPipelineForText2Image.from_pretrained(
-            config["repo_id"],
-            #variant="fp16",
-            #cache_dir=config["cache_dir"],
-            torch_dtype=torch.bfloat16
-        ).to("cuda")
+        pipe = b_pipe
         
     image = generate_image_with_progress(
         model_name,pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, seed=seed, progress=progress