Update app.py

app.py

@@ -12,12 +12,11 @@ from diffusers import DiffusionPipeline, EulerDiscreteScheduler
 # For example, you can replace with a local snapshot inside the Space repo.
 MODEL_ID = os.getenv("MODEL_ID", "Tongyi-MAI/Z-Image-Turbo")
 
-dtype = torch.float16 if torch.cuda.is_available() else torch.float32
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
 pipe = DiffusionPipeline.from_pretrained(
     MODEL_ID,
-    torch_dtype=dtype,
+    torch_dtype=torch.bfloat16,
     use_safetensors=True,
 )