From db9a1fe1c5ed4e7bc5043ddb537f3482c65615eb Mon Sep 17 00:00:00 2001
From: Baolin Peng <Baolin@users.noreply.huggingface.co>
Date: Tue, 25 Oct 2022 16:33:07 +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 2db99e0..fd80e1b 100644
--- a/README.md
+++ b/README.md
@@ -33,7 +33,7 @@ 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-base-seq2seq")
 model = AutoModelForSeq2SeqLM.from_pretrained("microsoft/GODEL-v1_1-base-seq2seq")