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