Update app.py

app.py

@@ -77,7 +77,7 @@ class ModelWrapper:
     def _get_time():
         return time.time()
         
-    @spaces.GPU(duration=100)
+    @spaces.GPU()
     def sample(self, noise, unet_added_conditions, prompt_embed, fast_vae_decode):
         alphas_cumprod = self.scheduler.alphas_cumprod.to(self.device)
 
@@ -90,7 +90,7 @@ class ModelWrapper:
         else:
             raise NotImplementedError()
 
-        noise = noise.to(device="cuda", dtype=torch.float16)
+        noise = noise.to(device="cuda", dtype=torch.float32)
         print(f'noise: {noise.dtype}')
         DTYPE = prompt_embed.dtype
         print(f'prompt_embed: {DTYPE}')