summaryrefslogtreecommitdiff
path: root/common
diff options
context:
space:
mode:
Diffstat (limited to 'common')
-rw-r--r--common/sampling.h2
1 files changed, 2 insertions, 0 deletions
diff --git a/common/sampling.h b/common/sampling.h
index 99fb07ac..4fc86595 100644
--- a/common/sampling.h
+++ b/common/sampling.h
@@ -11,6 +11,7 @@
// sampler types
enum class llama_sampler_type : char {
+ DRY ='d',
TOP_K = 'k',
TOP_P = 'p',
MIN_P = 'm',
@@ -53,6 +54,7 @@ typedef struct llama_sampling_params {
llama_sampler_type::TYPICAL_P,
llama_sampler_type::TOP_P,
llama_sampler_type::MIN_P,
+ llama_sampler_type::TOP_N_SIGMA,
llama_sampler_type::TEMPERATURE
};