mirror of
https://github.com/ggml-org/llama.cpp.git
synced 2026-09-13 02:10:45 +02:00
Add a numeric context parameter declaring the maximum outputs one sequence
This commit is contained in:
@@ -2382,6 +2382,11 @@ int32_t common_speculative_n_outputs_max(int32_t n_batch, int32_t n_parallel, in
|
||||
return std::min<int64_t>(n_batch, n_outputs);
|
||||
}
|
||||
|
||||
int32_t common_speculative_n_outputs_per_seq_max(int32_t n_batch, int32_t n_draft) {
|
||||
const int64_t n_outputs = 1 + (int64_t) std::max(0, n_draft);
|
||||
return std::min<int64_t>(n_batch, n_outputs);
|
||||
}
|
||||
|
||||
// initialization of the speculative decoding system
|
||||
//
|
||||
common_speculative * common_speculative_init(common_params_speculative & params, uint32_t n_seq) {
|
||||
|
||||
Reference in New Issue
Block a user