fix:ollama text embedding 500 error (langgenius#8252)

cuiks · Sep 26, 2024 · 1d3f5e3 · 1d3f5e3
1 parent 5ab0ba6
commit 1d3f5e3
Showing 1 changed file with 1 addition and 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
@@ -77,7 +77,7 @@ def _invoke(
                 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))