From acf5d02104a34927c0c660b9d0f61f0855a720e1 Mon Sep 17 00:00:00 2001 From: josc146 Date: Thu, 14 Mar 2024 12:24:45 +0800 Subject: [PATCH] update global_penalty desc --- backend-python/utils/rwkv.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend-python/utils/rwkv.py b/backend-python/utils/rwkv.py index 0429037..96ca8fe 100644 --- a/backend-python/utils/rwkv.py +++ b/backend-python/utils/rwkv.py @@ -677,7 +677,10 @@ class ModelConfigBody(BaseModel): frequency_penalty: float = Field(default=None, ge=-2, le=2) penalty_decay: float = Field(default=None, ge=0.99, le=0.999) 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 = { "json_schema_extra": {