Update model.py

model.py

@@ -1125,7 +1125,7 @@ class StreamMultiDiffusion(nn.Module):
         model_pred = self.unet(
             x_t_latent_plus_uc.to(self.dtype),  # (B, 4, h, w)
             t_list,  # (B,)
-            encoder_hidden_states=self.prompt_embeds,  # (B, 77, 768)
+            encoder_hidden_states=self.prompt_embeds.float(),  # (B, 77, 768)
             return_dict=False,
         )[0]  # (B, 4, h, w)
         print('222222222222222', model_pred.dtype)