diff --git a/LLama.KernelMemory/LLamaSharpTextEmbeddingGenerator.cs b/LLama.KernelMemory/LLamaSharpTextEmbeddingGenerator.cs
index 7806282d..0451d5bf 100644
--- a/LLama.KernelMemory/LLamaSharpTextEmbeddingGenerator.cs
+++ b/LLama.KernelMemory/LLamaSharpTextEmbeddingGenerator.cs
@@ -27,7 +27,7 @@ namespace LLamaSharp.KernelMemory
         public LLamaSharpTextEmbeddingGenerator(LLamaSharpConfig config)
         {
             this._config = config;
-            var @params = new ModelParams(_config.ModelPath);
+            var @params = new ModelParams(_config.ModelPath) { EmbeddingMode = true };
             _weights = LLamaWeights.LoadFromFile(@params);
             _embedder = new LLamaEmbedder(_weights, @params);
             _ownsWeights = true;
@@ -42,7 +42,7 @@ namespace LLamaSharp.KernelMemory
         public LLamaSharpTextEmbeddingGenerator(LLamaSharpConfig config, LLamaWeights weights)
         {
             this._config = config;
-            var @params = new ModelParams(_config.ModelPath);
+            var @params = new ModelParams(_config.ModelPath) { EmbeddingMode = true };
             _weights = weights;
             _embedder = new LLamaEmbedder(_weights, @params);
             _ownsEmbedder = true;