diff --git a/LLama.KernelMemory/BuilderExtensions.cs b/LLama.KernelMemory/BuilderExtensions.cs
index 7afac4bb..27ede7aa 100644
--- 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);