From 07a3b7a7c93355f49367e15432ddc4d8331cee2b Mon Sep 17 00:00:00 2001
From: Baolin Peng <Baolin@users.noreply.huggingface.co>
Date: Tue, 25 Oct 2022 16:32:44 +0000
Subject: [PATCH] Update README.md

---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index b90e22f..bddf574 100644
--- a/README.md
+++ b/README.md
@@ -32,7 +32,7 @@ ArXiv paper: [https://arxiv.org/abs/2206.11309](https://arxiv.org/abs/2206.11309
 Now we are ready to try out how the model works as a chatting partner!
 
 ```python
-from transformers import AutoTokenizer,AutoModel
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 tokenizer = AutoTokenizer.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-large-seq2seq")
 def generate(instruction, knowledge, dialog):