diff --git a/LLama/LLamaContext.cs b/LLama/LLamaContext.cs index 2902dc8f..db0ac179 100644 --- a/LLama/LLamaContext.cs +++ b/LLama/LLamaContext.cs @@ -86,6 +86,15 @@ namespace LLama NativeHandle = SafeLLamaContextHandle.Create(model.NativeHandle, lparams); } + /// + /// Set the seed for the RNG + /// + /// + public void SetSeed(uint seed) + { + NativeHandle.SetSeed(seed); + } + /// /// Tokenize a string. /// diff --git a/LLama/Native/SafeLLamaContextHandle.cs b/LLama/Native/SafeLLamaContextHandle.cs index bb49cc0f..df33076f 100644 --- a/LLama/Native/SafeLLamaContextHandle.cs +++ b/LLama/Native/SafeLLamaContextHandle.cs @@ -283,5 +283,14 @@ namespace LLama.Native } } #endregion + + /// + /// Set the RNG seed + /// + /// + public void SetSeed(uint seed) + { + NativeApi.llama_set_rng_seed(this, seed); + } } }