From 6473f8d5e518193d4ee4e415bf9c4cb5ad529a85 Mon Sep 17 00:00:00 2001 From: Martin Evans Date: Thu, 10 Aug 2023 17:51:51 +0100 Subject: [PATCH] Temporarily added a `Console.WriteLine` into the test, to print the embedding vector for "cat" in CI --- LLama.Unittest/LLamaEmbedderTests.cs | 2 ++ LLama/LLamaEmbedder.cs | 4 ++-- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/LLama.Unittest/LLamaEmbedderTests.cs b/LLama.Unittest/LLamaEmbedderTests.cs index 1c4b9fd7..77315310 100644 --- a/LLama.Unittest/LLamaEmbedderTests.cs +++ b/LLama.Unittest/LLamaEmbedderTests.cs @@ -35,6 +35,8 @@ namespace LLama.Unittest var kitten = _embedder.GetEmbeddings("kitten"); var spoon = _embedder.GetEmbeddings("spoon"); + Console.WriteLine(string.Join(",", cat)); + var close = Dot(cat, kitten); var far = Dot(cat, spoon); diff --git a/LLama/LLamaEmbedder.cs b/LLama/LLamaEmbedder.cs index 6b82c4d8..5acf756b 100644 --- a/LLama/LLamaEmbedder.cs +++ b/LLama/LLamaEmbedder.cs @@ -53,7 +53,7 @@ namespace LLama { threads = Math.Max(Environment.ProcessorCount / 2, 1); } - int n_past = 0; + if (addBos) { text = text.Insert(0, " "); @@ -65,7 +65,7 @@ namespace LLama if (embed_inp_array.Length > 0) { - if (NativeApi.llama_eval(_ctx, embed_inp_array, embed_inp_array.Length, n_past, threads) != 0) + if (NativeApi.llama_eval(_ctx, embed_inp_array, embed_inp_array.Length, 0, threads) != 0) { throw new RuntimeError("Failed to eval."); }