added signatre based check

check_app.py

@@ -46,7 +46,14 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
         progress(cur_prg, desc=f"Step {step_index}/{num_steps}")
         return callback_kwargs
     print(f"START GENR ")
-    if hasattr(pipe, "guidance_scale") and hasattr(pipe, "callback_on_step_end"):
+    # Get the signature of the pipe
+    pipe_signature = signature(pipe)
+    
+    # Check for the presence of "guidance_scale" and "callback_on_step_end" in the signature
+    has_guidance_scale = "guidance_scale" in pipe_signature.parameters
+    has_callback_on_step_end = "callback_on_step_end" in pipe_signature.parameters
+    
+    if has_guidance_scale and has_callback_on_step_end:
         print("has callback_on_step_end and has guidance_scale")
         image = pipe(
             prompt,
@@ -55,7 +62,7 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
             guidance_scale=guidance_scale,
             callback_on_step_end=callback,
         ).images[0]
-    elif not hasattr(pipe, "callback_on_step_end") and hasattr(pipe, "guidance_scale"):
+    elif not has_callback_on_step_end and has_guidance_scale:
         print("NO callback_on_step_end and has guidance_scale")
         image = pipe(
             prompt,
@@ -63,15 +70,15 @@ def generate_image_with_progress(pipe, prompt, num_steps, guidance_scale=None, s
             guidance_scale=guidance_scale,
             generator=generator,
         ).images[0]
-    elif hasattr(pipe, "callback_on_step_end") and not hasattr(pipe, "guidance_scale"):
-        print(" has callback_on_step_end and NO guidance_scale")
+    elif has_callback_on_step_end and not has_guidance_scale:
+        print("has callback_on_step_end and NO guidance_scale")
         image = pipe(
             prompt,
             num_inference_steps=num_steps,
             generator=generator,
-            callback_on_step_end=callback
+            callback_on_step_end=callback,
         ).images[0]
-    elif not hasattr(pipe, "callback_on_step_end") and not hasattr(pipe, "guidance_scale"):
+    elif not has_callback_on_step_end and not has_guidance_scale:
         print("NO callback_on_step_end and NO guidance_scale")
         image = pipe(
             prompt,
@@ -102,7 +109,7 @@ def create_pipeline_logic(prompt_text, model_name):
             #cache_dir=config["cache_dir"],
             torch_dtype=torch.bfloat16
         ).to("cuda")
-        print(f"PIPE ATTR {signature(pipe)}")
+        
     image = generate_image_with_progress(
         pipe, prompt_text, num_steps=num_steps, guidance_scale=guidance_scale, seed=seed, progress=progress
     )