Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions examples/embedding/embedding.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -283,10 +283,6 @@ int main(int argc, char ** argv) {
float * out = emb + e * n_embd;
batch_decode(ctx, batch, out, s, n_embd, params.embd_normalize);

if (params.embd_out == "raw") {
print_raw_embeddings(emb, n_embd_count, n_embd, model, pooling_type, params.embd_normalize);
}

if (params.embd_out.empty()) {
LOG("\n");

Expand Down Expand Up @@ -402,6 +398,10 @@ int main(int argc, char ** argv) {
if (notArray) LOG("\n}\n");
}

if (params.embd_out == "raw") {
print_raw_embeddings(emb, n_embd_count, n_embd, model, pooling_type, params.embd_normalize);
}

LOG("\n");
llama_perf_context_print(ctx);

Expand Down
Loading