Update app.py

app.py

@@ -123,7 +123,8 @@ class ModelWrapper:
 
         add_time_ids = self.build_condition_input(height, width).repeat(num_images, 1)
 
-        noise = torch.randn(num_images, 4, height // self.vae_downsample_ratio, width // self.vae_downsample_ratio, generator=generator).to(device="cuda")
+        noise = torch.randn(num_images, 4, height // self.vae_downsample_ratio, width // self.vae_downsample_ratio, generator=generator).to(device="cuda",dtype=torch.float16)
+        
         noise = noise.to(torch.float16)
 
         prompt_inputs = self._encode_prompt(prompt)