fix: ollama text embedding 500 error (#8131)

langgenius · Sep 9, 2024 · a27d4d5 · a27d4d5
1 parent 50d92f0
commit a27d4d5
Showing 1 changed file with 2 additions 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
@@ -89,7 +89,8 @@ def _invoke(self, model: str, credentials: dict,
             endpoint_url,
             headers=headers,
             data=json.dumps(payload),
-            timeout=(10, 300)
+            timeout=(10, 300),
+            options={"use_mmap": "true"}
         )
 
         response.raise_for_status()  # Raise an exception for HTTP errors