From: Georgi Gerganov <redacted>
Date: Thu, 14 Mar 2024 10:37:20 +0000 (+0200)
Subject: embedding : print all resulting embeddings (#899)
X-Git-Tag: upstream/0.0.4488~2068
X-Git-Url: https://git.djapps.eu/?a=commitdiff_plain;h=68265ebfc6a1bed022973ea0c3145be1450b7e70;p=pkg%2Fggml%2Fsources%2Fllama.cpp

embedding : print all resulting embeddings (#899)
---

diff --git a/examples/embedding/embedding.cpp b/examples/embedding/embedding.cpp
index f390c406..895469a3 100644
--- a/examples/embedding/embedding.cpp
+++ b/examples/embedding/embedding.cpp
@@ -167,9 +167,9 @@ int main(int argc, char ** argv) {
     float * out = emb + p * n_embd;
     batch_decode(ctx, batch, out, s, n_embd);
 
-    // print first 3 embeddings
+    // print the first part of the embeddings
     fprintf(stdout, "\n");
-    for (int j = 0; j < std::min(3, n_prompts); j++) {
+    for (int j = 0; j < n_prompts; j++) {
         fprintf(stdout, "embedding %d: ", j);
         for (int i = 0; i < std::min(16, n_embd); i++) {
             fprintf(stdout, "%f ", emb[j * n_embd + i]);