mirror of https://github.com/InternLM/InternLM
set the value of repetition_penalty to 1.0 to avoid random outputs (#99)
parent
e746754b6e
commit
1ebc3cca99
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue