From ad6c6a4762ef4a3364d42f0a8850b5057446daa1 Mon Sep 17 00:00:00 2001
From: tpoisonooo <khj.application@aliyun.com>
Date: Thu, 9 Nov 2023 11:45:22 +0800
Subject: [PATCH] Update README.md

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

diff --git a/README.md b/README.md
index 31643ae2..8475fdcf 100644
--- a/README.md
+++ b/README.md
@@ -142,7 +142,7 @@ max_length = 256
 outputs = model.generate(
     inputs.input_ids.cuda(),
     max_length=max_length,
-    eos_token_id=tokenizer.eos_token_id
+    eos_token_id=tokenizer.eos_token_id,
     do_sample=True,
     repetition_penalty=1.3,
     no_repeat_ngram_size=5,