class LlamaSampler: append add_xtc() and add_top_n_sigma()

Author: JamePeng

llama_cpp/_internals.py

@@ -791,6 +791,10 @@ def add_typical(self, p: float, min_keep: int):
         sampler = llama_cpp.llama_sampler_init_typical(p, min_keep)
         self._add_sampler(sampler)
 
+    def add_xtc(self, p: float, t: float, min_keep: int, seed: int):
+        sampler = llama_cpp.llama_sampler_init_xtc(p, t, min_keep, seed)
+        self._add_sampler(sampler)
+
     def add_temp(self, temp: float):
         sampler = llama_cpp.llama_sampler_init_temp(temp)
         self._add_sampler(sampler)
@@ -799,6 +803,10 @@ def add_temp_ext(self, t: float, delta: float, exponent: float):
         sampler = llama_cpp.llama_sampler_init_temp_ext(t, delta, exponent)
         self._add_sampler(sampler)
 
+    def add_top_n_sigma(self, n: float):
+        sampler = llama_cpp.llama_sampler_init_top_n_sigma(n)
+        self._add_sampler(sampler)
+
     def add_mirostat(self, n_vocab: int, seed: int, tau: float, eta: float, m: int):
         sampler = llama_cpp.llama_sampler_init_mirostat(n_vocab, seed, tau, eta, m)
         self._add_sampler(sampler)