Update app.py

app.py

@@ -438,6 +438,7 @@ def remove_custom_lora(selected_indices, current_loras, gallery):
   
 def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress):
     print("Generating image...")
+    print("Debug: Starting image generation")
     pipe.to("cuda")
     generator = torch.Generator(device="cuda").manual_seed(seed)
     final_image = None  # Initialize final_image
@@ -454,7 +455,7 @@ def generate_image(prompt_mash, steps, seed, cfg_scale, width, height, progress)
             output_type="pil",
             good_vae=good_vae,
         ):
-            #print("Image generated successfully.")  # Debugging statement
+            print(f"Debug: Yielding image of type {type(img)}")  # Debugging
             yield img
             final_image = img  # Update final_image with the current image
         return final_image