github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

add repetition_penalty in GenerationConfig in web_demo.py (#48) 

Co-authored-by: YWMditto <862779238@qq.com>
pull/23/head^2
YWMditto 2023-07-14 17:03:52 +08:00 committed by GitHub
parent be50c02949
commit fda99947ad
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
web_demo.py
View File

@ -147,7 +147,8 @@ class GenerationConfig:
top_p: Optional[float] = None top_p: Optional[float] = None
temperature: Optional[float] = None temperature: Optional[float] = None
do_sample: Optional[bool] = True do_sample: Optional[bool] = True
repetition_penalty: Optional[float] = 1.1
@st.cache_resource @st.cache_resource
def load_model(): def load_model():