Update app.py

app.py

@@ -90,9 +90,9 @@ class ModelWrapper:
         else:
             raise NotImplementedError()
 
-        noise = noise.to(torch.float16)
+        noise = noise.to(torch.float32)
         print(f'noise: {noise.dtype}')
-        prompt_embed = prompt_embed.to(torch.float16)
+        prompt_embed = prompt_embed.to(torch.float32)
         DTYPE = prompt_embed.dtype
         print(f'prompt_embed: {DTYPE}')
         
@@ -170,7 +170,7 @@ def get_x0_from_noise(sample, model_output, alphas_cumprod, timestep):
     return pred_original_sample
 
 class SDXLTextEncoder(torch.nn.Module):
-    def __init__(self, model_id, revision, accelerator, dtype=torch.float16):
+    def __init__(self, model_id, revision, accelerator, dtype=torch.float32):
         super().__init__()
 
         self.text_encoder_one = CLIPTextModel.from_pretrained(model_id, subfolder="text_encoder", revision=revision).to(0).to(dtype=dtype)