mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-09-19 17:24:57 +02:00
return output ID from llama_batch_ext_add/set
This commit is contained in:
+1
-1
@@ -606,7 +606,7 @@ struct common_batch {
|
||||
}
|
||||
void set_logits_last() {
|
||||
if (!tokens.empty()) {
|
||||
llama_batch_ext_set_logits_last(batch.get());
|
||||
llama_batch_ext_set_output_last(batch.get());
|
||||
tokens.back().logits = true;
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user