Update app.py

app.py

@@ -14,7 +14,7 @@ pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 g_cuda = torch.Generator()
 
 def inference(prompt, negative_prompt, num_samples, height=512, width=512, num_inference_steps=50, guidance_scale=7.5, g_seed=52362):
-    with torch.autocast("cuda"), torch.inference_mode():
+    with torch.inference_mode():
         return pipe(
                 prompt, height=int(height), width=int(width),
                 negative_prompt=negative_prompt,