completed top nsigma sampler implementation
This commit is contained in:
parent
ddc3c2208a
commit
da038d8715
5 changed files with 112 additions and 79 deletions
|
|
@ -899,6 +899,13 @@ common_params_context common_params_parser_init(common_params & params, llama_ex
|
|||
params.sampling.min_p = std::stof(value);
|
||||
}
|
||||
).set_sparam());
|
||||
add_opt(common_arg(
|
||||
{"--top-nsigma"}, "N",
|
||||
string_format("top-n-sigma sampling (default: %d, -1 = disabled)", params.sampling.top_n_sigma),
|
||||
[](common_params & params, const std::string & value) {
|
||||
params.sampling.top_n_sigma = std::stof(value);
|
||||
}
|
||||
).set_sparam());
|
||||
add_opt(common_arg(
|
||||
{"--xtc-probability"}, "N",
|
||||
string_format("xtc probability (default: %.1f, 0.0 = disabled)", (double)params.sampling.xtc_probability),
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue