From fda99947aded9d4b47e20b8ad7e5e8aa22fc28f4 Mon Sep 17 00:00:00 2001
From: YWMditto <46778265+YWMditto@users.noreply.github.com>
Date: Fri, 14 Jul 2023 17:03:52 +0800
Subject: [PATCH] add repetition_penalty in GenerationConfig in web_demo.py
 (#48)

Co-authored-by: YWMditto <862779238@qq.com>
---
 web_demo.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/web_demo.py b/web_demo.py
index 59fe79f..2cdcac1 100644
--- a/web_demo.py
+++ b/web_demo.py
@@ -147,7 +147,8 @@ class GenerationConfig:
     top_p: Optional[float] = None
     temperature: Optional[float] = None
     do_sample: Optional[bool] = True
-    
+    repetition_penalty: Optional[float] = 1.1
+
 
 @st.cache_resource
 def load_model():