From 3fda708eaa649469602a070147fa93886d25224d Mon Sep 17 00:00:00 2001
From: Kenneth Tang <32535494+ChengYen-Tang@users.noreply.github.com>
Date: Tue, 19 Mar 2024 06:53:16 +0000
Subject: [PATCH] Fix System.ArgumentException: EmbeddingMode must be true

---
 LLama.KernelMemory/LLamaSharpTextEmbeddingGenerator.cs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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;