fix

app.py

@@ -95,8 +95,9 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
 
         try:
             print("Calling the transformer with latents")
-            # Dummy call to transformer to understand the shape requirement
-            _ = pipe.transformer(latents)
+            # Initialize timestep variable
+            timestep = torch.tensor([num_inference_steps], device=device, dtype=dtype)
+            _ = pipe.transformer(latents, timestep=timestep)
             print("Transformer call succeeded")
         except Exception as e:
             print(f"Transformer call failed with error: {e}")
@@ -120,9 +121,6 @@ def infer(prompt, init_image=None, seed=42, randomize_seed=False, width=1024, he
     print("Inference complete")
     return image, seed
 
-
-
-
 # Define example prompts
 examples = [
     "a tiny astronaut hatching from an egg on the moon",