sampling : temp == 0.0 -> no probs, temp < 0.0 -> probs
This commit is contained in:
parent
c86cca8061
commit
bbfc62ac2f
3 changed files with 5 additions and 4 deletions
|
@ -118,7 +118,7 @@ int main(int argc, char ** argv) {
|
|||
std::vector<seq_draft> drafts(n_seq_dft);
|
||||
|
||||
params.sparams.grammar.clear(); // the draft samplers will copy the target sampler's grammar
|
||||
params.sparams.temp = 0.0f;
|
||||
params.sparams.temp = -1.0f; // force greedy sampling with probs for the draft model
|
||||
|
||||
for (int s = 0; s < n_seq_dft; ++s) {
|
||||
drafts[s].ctx_sampling = llama_sampling_init(params.sparams);
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue