Skip to content

Commit ab34b2d

Browse files
authored
Bugfix/Update ollama embedding to use mmap true by default (#2862)
update ollama embedding to use mmap true by default
1 parent 417fe6b commit ab34b2d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

packages/components/nodes/embeddings/OllamaEmbedding/OllamaEmbedding.ts

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -61,6 +61,7 @@ class OllamaEmbedding_Embeddings implements INode {
6161
label: 'Use MMap',
6262
name: 'useMMap',
6363
type: 'boolean',
64+
default: true,
6465
optional: true,
6566
additionalParams: true
6667
}
@@ -83,7 +84,9 @@ class OllamaEmbedding_Embeddings implements INode {
8384
const requestOptions: OllamaInput = {}
8485
if (numThread) requestOptions.numThread = parseFloat(numThread)
8586
if (numGpu) requestOptions.numGpu = parseFloat(numGpu)
86-
if (useMMap !== undefined) requestOptions.useMMap = useMMap
87+
88+
// default useMMap to true
89+
requestOptions.useMMap = useMMap === undefined ? true : useMMap
8790

8891
if (Object.keys(requestOptions).length) obj.requestOptions = requestOptions
8992

0 commit comments

Comments
 (0)