From 906e058d229031a19cee6f82f789b293a11224af Mon Sep 17 00:00:00 2001
From: Omar Sanseviero <osanseviero@users.noreply.huggingface.co>
Date: Wed, 29 May 2024 12:27:10 +0000
Subject: [PATCH] Update README.md (#118)

- Update README.md (02bf128f36f0a0e3aace3c8f25fec1e07b468db7)
- Update README.md (21014479dba59d2ee7bd508cee9e2463eaf18732)


Co-authored-by: Matthew Carrigan <Rocketknight1@users.noreply.huggingface.co>
---
 README.md | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/README.md b/README.md
index cfaa7a0..523b6f8 100644
--- a/README.md
+++ b/README.md
@@ -316,26 +316,20 @@ messages = [
     {"role": "user", "content": "Who are you?"},
 ]
 
-prompt = pipeline.tokenizer.apply_chat_template(
-		messages, 
-		tokenize=False, 
-		add_generation_prompt=True
-)
-
 terminators = [
     pipeline.tokenizer.eos_token_id,
     pipeline.tokenizer.convert_tokens_to_ids("<|eot_id|>")
 ]
 
 outputs = pipeline(
-    prompt,
+    messages,
     max_new_tokens=256,
     eos_token_id=terminators,
     do_sample=True,
     temperature=0.6,
     top_p=0.9,
 )
-print(outputs[0]["generated_text"][len(prompt):])
+print(outputs[0]["generated_text"][-1])
 ```
 
 #### Transformers AutoModelForCausalLM