KernelMemory EmbeddingMode bug correction

2024-02-04 08:37:08 +01:00 · 2024-02-04 08:37:08 +01:00 · f578fcafa3
parent 2323988cc7
commit f578fcafa3
1 changed files with 2 additions and 1 deletions
--- a/LLama.KernelMemory/BuilderExtensions.cs
+++ b/LLama.KernelMemory/BuilderExtensions.cs
@ -80,7 +80,8 @@ namespace LLamaSharp.KernelMemory
            {
                ContextSize = config?.ContextSize ?? 2048,
                Seed = config?.Seed ?? 0,
-                GpuLayerCount = config?.GpuLayerCount ?? 20
+                GpuLayerCount = config?.GpuLayerCount ?? 20,
+                EmbeddingMode = true
            };
            var weights = LLamaWeights.LoadFromFile(parameters);
            var context = weights.CreateContext(parameters);