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

set the value of repetition_penalty to 1.0 to avoid random outputs (#99)

pull/106/head
Yang Gao 2023-07-17 23:19:48 +08:00 committed by GitHub
parent e746754b6e
commit 1ebc3cca99
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
web_demo.py
View File

@ -147,7 +147,7 @@ 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 repetition_penalty: Optional[float] = 1.0
@st.cache_resource @st.cache_resource