log shape and error handling

app.py

@@ -46,7 +46,7 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
         print("No initial image provided, processing text2img")
         # Process text2img
         try:
-            print("Calling the diffusion pipeline without latents")
+            print("Calling the diffusion pipeline for text2img")
             result = pipe(
                 prompt=prompt,
                 height=height,
@@ -56,10 +56,10 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
                 guidance_scale=0.0
             )
             image = result.images[0]
-            latents = result.latents
-            
-            # Log the latent shapes from text2img process
-            print(f"Latents shape from text2img: {latents.shape}")
+            print(f"Generated image shape: {image.size}")
+
+            # Since the 'latents' attribute is not present, we need to inspect other attributes
+            print(f"Result attributes: {dir(result)}")
         except Exception as e:
             print(f"Pipeline call failed with error: {e}")
             raise
@@ -104,24 +104,23 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
             raise
 
         print("Calling the diffusion pipeline with latents")
-        image = pipe(
-            prompt=prompt,
-            height=height,
-            width=width,
-            num_inference_steps=num_inference_steps,
-            generator=generator,
-            guidance_scale=0.0,
-            latents=latents
-        ).images[0]
+        try:
+            image = pipe(
+                prompt=prompt,
+                height=height,
+                width=width,
+                num_inference_steps=num_inference_steps,
+                generator=generator,
+                guidance_scale=0.0,
+                latents=latents
+            ).images[0]
+        except Exception as e:
+            print(f"Pipeline call with latents failed with error: {e}")
+            raise
     
     print("Inference complete")
     return image, seed
 
-
-
-
-
-
 # Define example prompts
 examples = [
     "a tiny astronaut hatching from an egg on the moon",