Skip to content

Commit 24b850b

Browse files
committed
Move raw output handling into else-if block with other format handlers
1 parent c667120 commit 24b850b

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

examples/embedding/embedding.cpp

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -396,9 +396,7 @@ int main(int argc, char ** argv) {
396396
}
397397

398398
if (notArray) LOG("\n}\n");
399-
}
400-
401-
if (params.embd_out == "raw") {
399+
} else if (params.embd_out == "raw") {
402400
print_raw_embeddings(emb, n_embd_count, n_embd, model, pooling_type, params.embd_normalize);
403401
}
404402

@@ -407,6 +405,11 @@ int main(int argc, char ** argv) {
407405

408406
// clean up
409407
llama_batch_free(batch);
408+
409+
410+
411+
412+
410413
llama_backend_free();
411414

412415
return 0;

0 commit comments

Comments
 (0)