From 53a5574080b2779a4c57d79185a48a8ab43c2aee Mon Sep 17 00:00:00 2001 From: josc146 Date: Sat, 2 Mar 2024 16:52:53 +0800 Subject: [PATCH] improve parameters controllable range --- backend-python/utils/rwkv.py | 2 +- frontend/src/pages/Chat.tsx | 4 ++-- frontend/src/pages/Completion.tsx | 4 ++-- frontend/src/pages/Composition.tsx | 4 ++-- frontend/src/pages/Configs.tsx | 4 ++-- 5 files changed, 9 insertions(+), 9 deletions(-) diff --git a/backend-python/utils/rwkv.py b/backend-python/utils/rwkv.py index 6a0f64a..2fff4ee 100644 --- a/backend-python/utils/rwkv.py +++ b/backend-python/utils/rwkv.py @@ -667,7 +667,7 @@ def RWKV(model: str, strategy: str, tokenizer: Union[str, None]) -> AbstractRWKV class ModelConfigBody(BaseModel): max_tokens: int = Field(default=None, gt=0, le=102400) - temperature: float = Field(default=None, ge=0, le=2) + temperature: float = Field(default=None, ge=0, le=3) top_p: float = Field(default=None, ge=0, le=1) presence_penalty: float = Field(default=None, ge=-2, le=2) frequency_penalty: float = Field(default=None, ge=-2, le=2) diff --git a/frontend/src/pages/Chat.tsx b/frontend/src/pages/Chat.tsx index 8804407..401408f 100644 --- a/frontend/src/pages/Chat.tsx +++ b/frontend/src/pages/Chat.tsx @@ -247,7 +247,7 @@ const SidePanel: FC = observer(() => { { commonStore.setChatParams({ @@ -258,7 +258,7 @@ const SidePanel: FC = observer(() => { { commonStore.setChatParams({ topP: data.value diff --git a/frontend/src/pages/Completion.tsx b/frontend/src/pages/Completion.tsx index 8a209d0..5b614c2 100644 --- a/frontend/src/pages/Completion.tsx +++ b/frontend/src/pages/Completion.tsx @@ -188,7 +188,7 @@ const CompletionPanel: FC = observer(() => { { setParams({ @@ -199,7 +199,7 @@ const CompletionPanel: FC = observer(() => { { setParams({ topP: data.value diff --git a/frontend/src/pages/Composition.tsx b/frontend/src/pages/Composition.tsx index 79425ae..43d872a 100644 --- a/frontend/src/pages/Composition.tsx +++ b/frontend/src/pages/Composition.tsx @@ -275,7 +275,7 @@ const CompositionPanel: FC = observer(() => { { setParams({ @@ -286,7 +286,7 @@ const CompositionPanel: FC = observer(() => { { setParams({ topP: data.value diff --git a/frontend/src/pages/Configs.tsx b/frontend/src/pages/Configs.tsx index e95a09f..62098ba 100644 --- a/frontend/src/pages/Configs.tsx +++ b/frontend/src/pages/Configs.tsx @@ -176,7 +176,7 @@ const Configs: FC = observer(() => { { setSelectedConfigApiParams({ @@ -187,7 +187,7 @@ const Configs: FC = observer(() => { { setSelectedConfigApiParams({ topP: data.value