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

View File

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