Files
ragflow/conf/models/nvidia.json
jay77721 2e2d12d262 feat: add batch_size to all embedding model configs (#17877)
## Summary

Add a `batch_size` field to every embedding model entry in
`conf/models/*.json`. The field represents the maximum number of text
inputs that can be submitted to the embedding API in a single request.

**75 embedding models across 30 config files** now carry a `batch_size`.
Values were verified against each provider's official documentation (see
the verification table at `Desktop/embedding_models_verified.md`).

## Distribution

| batch_size | # models | Provider / Model |
|---|---|---|
| 1 | 2 | AWS Bedrock `amazon.titan-embed-text-v1/v2:0` — Bedrock
`invoke` accepts a single input per call |
| 10 | 3 | Aliyun `text-embedding-v3/v4`, Volcengine
`doubao-embedding-vision-251215` |
| 16 | 5 | BaiChuan `Baichuan-Text-Embedding`, Baidu Qianfan
`embedding-v1`, Mistral `mistral-embed`, Replicate (x2) |
| 32 | 10 | NVIDIA NIM (x3), SILICONFLOW (x2), PPIO (x3), GiteeAI
`bge-m3`, HuaweiCloud `bge-m3` |
| 50 | 4 | Tencent Hunyuan `kinfra` embeddings (x4) — `InputList.N` max
50 |
| 96 | 8 | Cohere embed-v3/v4 (x5), Bedrock Cohere (x3) |
| 100 | 3 | Google Gemini `text-embedding-004`, Upstage (x2) |
| 512 | 5 | Zhipu GLM `embedding-2/3` (x2), Perplexity `pplx-embed`
(x2), Astraflow `text-embedding-3-large` |
| 1000 | 9 | Voyage AI (x9) — API reference max |
| 1024 | 1 | DeepInfra `Qwen/Qwen3-Embedding-4B` |
| 2048 | 15 | OpenAI (x3) + OpenAI-API-compatible proxies (CometAPI,
n1n, Jiekou.AI, GreenPT, TogetherAI, NovitaAI) — OpenAI contract limit |
| 16384 | 10 | Jina (x8), 302.AI, GiteeAI `jina-clip-v2` — no documented
Jina batch limit, safe high cap |

---------

Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
2026-08-06 10:47:38 +08:00

402 lines
7.8 KiB
JSON

{
"name": "NVIDIA",
"url": {
"default": "https://integrate.api.nvidia.com/v1"
},
"url_suffix": {
"chat": "chat/completions",
"models": "models",
"embedding": "embeddings",
"rerank": "ranking"
},
"class": "nvidia",
"models": [
{
"name": "deepseek-ai/deepseek-v4-flash",
"content_length": 1048576,
"max_output": 393216,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "deepseek-ai/deepseek-v4-pro",
"content_length": 1048576,
"max_output": 393216,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "google/diffusiongemma-26b-a4b-it",
"content_length": 262144,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "google/gemma-4-31b-it",
"content_length": 262144,
"max_output": 262144,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "meta/llama-3.1-70b-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "meta/llama-3.1-8b-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "meta/llama-3.2-11b-vision-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat",
"vision"
]
},
{
"name": "meta/llama-3.2-1b-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "meta/llama-3.2-3b-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "meta/llama-3.2-90b-vision-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat",
"vision"
],
"tools": {
"support": true
}
},
{
"name": "meta/llama-3.3-70b-instruct",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "meta/llama-guard-4-12b",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "minimaxai/minimax-m3",
"content_length": 1048576,
"max_output": 16384,
"model_types": [
"chat"
]
},
{
"name": "mistralai/mistral-medium-3.5-128b",
"content_length": 262144,
"max_output": 262144,
"model_types": [
"chat",
"vision"
],
"tools": {
"support": true
}
},
{
"name": "mistralai/mistral-nemotron",
"content_length": 131072,
"max_output": 131072,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "moonshotai/kimi-k2.6",
"content_length": 262144,
"max_output": 131072,
"model_types": [
"chat",
"vision"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/ising-calibration-1.5-31b",
"content_length": 8192,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "nvidia/llama-3.1-nemotron-nano-8b-v1",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/llama-3.1-nemotron-nano-vl-8b-v1",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat",
"vision"
]
},
{
"name": "nvidia/llama-3.1-nemotron-safety-guard-8b-v3",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/llama-3.3-nemotron-super-49b-v1",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/llama-3.3-nemotron-super-49b-v1.5",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"thinking": {
"default_value": true,
"clear_thinking": true
},
"tools": {
"support": true
}
},
{
"name": "nvidia/nemotron-3-embed-1b",
"max_tokens": 8192,
"model_types": [
"embedding"
],
"batch_size": 32
},
{
"name": "nvidia/nemotron-3-nano-30b-a3b",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/nemotron-3-nano-omni-30b-a3b-reasoning",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat",
"vision"
],
"thinking": {
"default_value": true,
"clear_thinking": true
},
"tools": {
"support": true
}
},
{
"name": "nvidia/nemotron-3-super-120b-a12b",
"content_length": 262144,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/nemotron-3-ultra-550b-a55b",
"content_length": 262144,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "nvidia/nemotron-3.5-content-safety",
"content_length": 8192,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "nvidia/nemotron-mini-4b-instruct",
"content_length": 8192,
"max_output": 4096,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/nemotron-nano-12b-v2-vl",
"content_length": 8192,
"max_output": 8192,
"model_types": [
"chat",
"vision"
]
},
{
"name": "nvidia/nv-embed-v1",
"max_tokens": 32768,
"model_types": [
"embedding"
],
"batch_size": 32
},
{
"name": "nvidia/nv-embedcode-7b-v1",
"max_tokens": 8192,
"model_types": [
"embedding"
],
"batch_size": 32
},
{
"name": "nvidia/nvidia-nemotron-nano-9b-v2",
"content_length": 131072,
"max_output": 8192,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "nvidia/riva-translate-4b-instruct-v2",
"content_length": 8192,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "openai/gpt-oss-120b",
"content_length": 131072,
"max_output": 131072,
"model_types": [
"chat"
],
"tools": {
"support": true
}
},
{
"name": "openai/gpt-oss-20b",
"content_length": 131072,
"max_output": 131072,
"model_types": [
"chat"
]
},
{
"name": "poolside/laguna-xs-2.1",
"content_length": 262144,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "stepfun-ai/step-3.7-flash",
"content_length": 32768,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "thinkingmachines/inkling",
"content_length": 8192,
"max_output": 8192,
"model_types": [
"chat"
]
},
{
"name": "z-ai/glm-5.2",
"content_length": 1000000,
"max_output": 128000,
"model_types": [
"chat"
]
}
]
}