Update app.py

app.py

@@ -94,6 +94,7 @@ class ModelWrapper:
         for constant in all_timesteps:
             current_timesteps = torch.ones(len(prompt_embed), device="cuda", dtype=torch.long) * constant
             eval_images = self.model(noise, current_timesteps, prompt_embed, added_cond_kwargs=unet_added_conditions).sample
+            print(typeof(eval_images))
 
             eval_images = get_x0_from_noise(noise, eval_images, alphas_cumprod, current_timesteps).to(self.DTYPE)
 
@@ -138,7 +139,7 @@ class ModelWrapper:
         }
 
         eval_images = self.sample(noise=noise, unet_added_conditions=unet_added_conditions, prompt_embed=batch_prompt_embeds, fast_vae_decode=fast_vae_decode)
-
+        print(typeof(eval_images))
         end_time = self._get_time()
 
         output_image_list = []