Update README.md

README.md

@@ -92,7 +92,7 @@ if torch.cuda.is_available():
 elif torch.backends.mps.is_available():
     # Apple Silicon
     device = "mps"
-    dtype = torch.float16
+    dtype = torch.bfloat16
 else:
     # CPU fallback (functional but very slow for this model)
     device = "cpu"
@@ -104,6 +104,8 @@ pipe = ZImagePipeline.from_pretrained(
     low_cpu_mem_usage=True,
 )
 
+pipe.enable_attention_slicing()
+
 if device == "cuda":
     pipe.enable_model_cpu_offload()
 else: