Update app.py

app.py

@@ -92,7 +92,7 @@ class ModelWrapper:
 
         #noise = noise.to(torch.float16)
         print(f'noise: {noise.dtype}')
-        prompt_embed = prompt_embed.to(torch.float32)
+        #prompt_embed = prompt_embed.to(torch.float32)
         DTYPE = prompt_embed.dtype
         print(f'prompt_embed: {DTYPE}')
         
@@ -140,7 +140,7 @@ class ModelWrapper:
         )
 
         unet_added_conditions = {
-            "time_ids": add_time_ids,
+            "time_ids": add_time_ids.to(torch.float32),
             "text_embeds": batch_pooled_prompt_embeds.squeeze(1)
         }