From 53f37a67049a5642bede016e266980b302c7c69e Mon Sep 17 00:00:00 2001 From: HowardChan Date: Wed, 11 Sep 2024 16:23:19 +0800 Subject: [PATCH] fix:ollama text embedding 500 error (#8252) --- .../model_providers/ollama/text_embedding/text_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py b/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py index ffa9ea2f90..2cfb79b241 100644 --- a/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py +++ b/api/core/model_runtime/model_providers/ollama/text_embedding/text_embedding.py @@ -77,7 +77,7 @@ class OllamaEmbeddingModel(TextEmbeddingModel): inputs.append(text) # Prepare the payload for the request - payload = {"input": inputs, "model": model, "options": {"use_mmap": "true"}} + payload = {"input": inputs, "model": model, "options": {"use_mmap": True}} # Make the request to the Ollama API response = requests.post(endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300))