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
+            current_timesteps = current_timesteps.to(torch.float32)
             print(current_timesteps.dtype)
             eval_images = self.model(noise, current_timesteps, prompt_embed, added_cond_kwargs=unet_added_conditions).sample
             print(type(eval_images))