Browse Source

Temporarily added a `Console.WriteLine` into the test, to print the embedding vector for "cat" in CI

tags/v0.5.1
Martin Evans 2 years ago
parent
commit
6473f8d5e5
2 changed files with 4 additions and 2 deletions
  1. +2
    -0
      LLama.Unittest/LLamaEmbedderTests.cs
  2. +2
    -2
      LLama/LLamaEmbedder.cs

+ 2
- 0
LLama.Unittest/LLamaEmbedderTests.cs View File

@@ -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);



+ 2
- 2
LLama/LLamaEmbedder.cs View File

@@ -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.");
}


Loading…
Cancel
Save