theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update global_penalty desc

Browse Source
This commit is contained in:
josc146 2024-03-14 12:24:45 +08:00
parent bf58841f00
commit acf5d02104
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
backend-python/utils/rwkv.py
View File

@ -677,7 +677,10 @@ class ModelConfigBody(BaseModel):
frequency_penalty: float = Field(default=None, ge=-2, le=2) frequency_penalty: float = Field(default=None, ge=-2, le=2)
penalty_decay: float = Field(default=None, ge=0.99, le=0.999) penalty_decay: float = Field(default=None, ge=0.99, le=0.999)
top_k: int = Field(default=None, ge=0, le=25) top_k: int = Field(default=None, ge=0, le=25)
global_penalty: bool = Field(default=None) global_penalty: bool = Field(
default=None,
description="When generating a response, whether to include the submitted prompt as a penalty factor. By turning this off, you will get the same generated results as official RWKV Gradio. If you find duplicate results in the generated results, turning this on can help avoid generating duplicates.",
)
model_config = { model_config = {
"json_schema_extra": { "json_schema_extra": {