Update app.py

app.py

@@ -257,7 +257,7 @@ if torch.cuda.is_available():
         model_path,
         #vae=vae, 内置VAE
         use_safetensors=True,
-        torch_dtype=torch.float16,
+        torch_dtype=torch.bfloat16,
     )
     pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
     pipe.to("cuda")