diff --git a/README.md b/README.md
index 3d58fbf..8bd2ed8 100644
--- a/README.md
+++ b/README.md
@@ -287,7 +287,7 @@ pipeline = transformers.pipeline(
     "text-generation",
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
-    device="auto",
+    device_map="auto",
 )
 
 messages = [