Update model.py

model.py

@@ -1120,6 +1120,7 @@ class StreamMultiDiffusion(nn.Module):
         else:
             x_t_latent_plus_uc = x_t_latent  # (T * p, 4, h, w)
 
+        print(x_t_latent_plus_uc.dtype, t_list.dtype, self.prompt_embeds.dtype, self.masks.dtype)
         model_pred = self.unet(
             x_t_latent_plus_uc,  # (B, 4, h, w)
             t_list,  # (B,)