From 74fedae5f7f3033584edbcda2ee2d0d48e99c647 Mon Sep 17 00:00:00 2001
From: Pedro Cuenca <pcuenq@users.noreply.huggingface.co>
Date: Tue, 23 Apr 2024 11:33:24 +0000
Subject: [PATCH] Fix typo in pipeline device argument (#48)

- Fix typo in pipeline device argument (346ad49926472b69607b89966a270af1da03fe57)


Co-authored-by: Chandan Singh <csinva@users.noreply.huggingface.co>
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = [