diff --git a/src/pages/playground/components/ground-embedding.tsx b/src/pages/playground/components/ground-embedding.tsx index 9ecce67c..e4603e50 100644 --- a/src/pages/playground/components/ground-embedding.tsx +++ b/src/pages/playground/components/ground-embedding.tsx @@ -352,7 +352,7 @@ const GroundEmbedding: React.FC = forwardRef((props, ref) => { ); diff --git a/src/pages/playground/components/ground-reranker.tsx b/src/pages/playground/components/ground-reranker.tsx index dc6ebb25..d3e637ca 100644 --- a/src/pages/playground/components/ground-reranker.tsx +++ b/src/pages/playground/components/ground-reranker.tsx @@ -416,7 +416,7 @@ const GroundReranker: React.FC = forwardRef((props, ref) => { ); diff --git a/src/pages/playground/hooks/use-init-meta.ts b/src/pages/playground/hooks/use-init-meta.ts index b1d07cfc..2b33abb7 100644 --- a/src/pages/playground/hooks/use-init-meta.ts +++ b/src/pages/playground/hooks/use-init-meta.ts @@ -77,7 +77,7 @@ export const useInitLLmMeta = ( let max_tokens: number = 0; if (n_ctx && n_slot) { - max_tokens = _.divide(n_ctx, n_slot); + max_tokens = _.floor(_.divide(n_ctx, n_slot)); } else if (max_model_len) { max_tokens = max_model_len; } else if (max_total_tokens) { @@ -86,7 +86,7 @@ export const useInitLLmMeta = ( return { max_tokens: max_tokens || 16 * 1024, - defaultFormValue: max_tokens ? _.divide(max_tokens, 2) : 1024 + defaultFormValue: max_tokens ? _.floor(_.divide(max_tokens, 2)) : 1024 }; };