Browse Source

Embeddings correction

pull/674/head
Zoli Somogyi 1 year ago
parent
commit
5d442fa2f8
1 changed files with 7 additions and 3 deletions
  1. +7
    -3
      LLama/LLamaEmbedder.cs

+ 7
- 3
LLama/LLamaEmbedder.cs View File

@@ -97,9 +97,13 @@ namespace LLama

private float[] GetEmbeddingsArray()
{
var embeddings = NativeApi.llama_get_embeddings_seq(Context.NativeHandle, LLamaSeqId.Zero);
if (embeddings.Length == 0)
return Array.Empty<float>();
var embeddings = NativeApi.llama_get_embeddings(Context.NativeHandle);
if (embeddings == null || embeddings.Length == 0)
{
embeddings = NativeApi.llama_get_embeddings_seq(Context.NativeHandle, LLamaSeqId.Zero);
if (embeddings == null || embeddings.Length == 0)
return Array.Empty<float>();
}

return embeddings.ToArray();
}


Loading…
Cancel
Save