Update app.py

app.py

@@ -90,6 +90,10 @@ class ModelWrapper:
             raise NotImplementedError()
         
         DTYPE = prompt_embed.dtype
+        print(DTYPE)
+        print(typeof(noise))
+        print(typeof(current_timesteps))
+        print(typeof(unet_added_conditions))
         
         for constant in all_timesteps:
             current_timesteps = torch.ones(len(prompt_embed), device="cuda", dtype=torch.long) * constant
@@ -139,7 +143,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 = []