Update app.py

app.py

@@ -90,9 +90,9 @@ class ModelWrapper:
         else:
             raise NotImplementedError()
 
-        noise = noise.to(torch.float32)
+        noise = noise.to(torch.float16)
         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}')