From fd38e1cf1592f2cdc58dbe94337d586fa2e8d5ea Mon Sep 17 00:00:00 2001 From: Joshua <138381132+joshua20231026@users.noreply.github.com> Date: Mon, 15 Apr 2024 13:50:16 +0800 Subject: [PATCH] nvidia-label-update (#3482) --- api/core/model_runtime/model_providers/nvidia/nvidia.yaml | 2 +- .../model_providers/nvidia/rerank/rerank-qa-mistral-4b.yaml | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/api/core/model_runtime/model_providers/nvidia/nvidia.yaml b/api/core/model_runtime/model_providers/nvidia/nvidia.yaml index c3c316321e..4d6da913c1 100644 --- a/api/core/model_runtime/model_providers/nvidia/nvidia.yaml +++ b/api/core/model_runtime/model_providers/nvidia/nvidia.yaml @@ -1,6 +1,6 @@ provider: nvidia label: - en_US: NVIDIA + en_US: API Catalog icon_small: en_US: icon_s_en.svg icon_large: diff --git a/api/core/model_runtime/model_providers/nvidia/rerank/rerank-qa-mistral-4b.yaml b/api/core/model_runtime/model_providers/nvidia/rerank/rerank-qa-mistral-4b.yaml index 7703ca21ab..461f4e1cbe 100644 --- a/api/core/model_runtime/model_providers/nvidia/rerank/rerank-qa-mistral-4b.yaml +++ b/api/core/model_runtime/model_providers/nvidia/rerank/rerank-qa-mistral-4b.yaml @@ -1,4 +1,4 @@ model: nv-rerank-qa-mistral-4b:1 model_type: rerank model_properties: - context_size: 8192 + context_size: 512