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);
+ }
}
}