Update app.py

app.py

@@ -140,7 +140,7 @@ class ModelWrapper:
         )
 
         unet_added_conditions = {
-            "time_ids": add_time_ids.torch.float32,
+            "time_ids": add_time_ids.to(torch.float32),
             "text_embeds": batch_pooled_prompt_embeds.squeeze(1)
         }