root/llama.cpp
1
0
Fork 0
mirror of https://github.com/ggerganov/llama.cpp.git synced 2024-12-25 19:04:35 +00:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

embedding : print all resulting embeddings (#899)

Browse Source
This commit is contained in:
Georgi Gerganov 2024-03-14 12:37:20 +02:00
parent 381da2d9f0
commit 68265ebfc6
No known key found for this signature in database
GPG Key ID: 449E073F9DC10735
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
examples/embedding/embedding.cpp
View File

@ -167,9 +167,9 @@ int main(int argc, char ** argv) {
float * out = emb + p * n_embd; float * out = emb + p * n_embd;
batch_decode(ctx, batch, out, s, n_embd); batch_decode(ctx, batch, out, s, n_embd);
// print first 3 embeddings // print the first part of the embeddings
fprintf(stdout, "\n"); 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); fprintf(stdout, "embedding %d: ", j);
for (int i = 0; i < std::min(16, n_embd); i++) { for (int i = 0; i < std::min(16, n_embd); i++) {
fprintf(stdout, "%f ", emb[j * n_embd + i]); fprintf(stdout, "%f ", emb[j * n_embd + i]);