diff --git a/LLama/Native/LLamaContextParams.cs b/LLama/Native/LLamaContextParams.cs
index 4e782cc9..0fd0a51a 100644
--- a/LLama/Native/LLamaContextParams.cs
+++ b/LLama/Native/LLamaContextParams.cs
@@ -18,6 +18,10 @@ namespace LLama.Native
///
public int n_parts;
///
+ /// number of layers to store in VRAM
+ ///
+ public int n_gpu_layers;
+ ///
/// RNG seed, -1 for random
///
public int seed;
diff --git a/LLama/Utils.cs b/LLama/Utils.cs
index 07172da2..2e2c992f 100644
--- a/LLama/Utils.cs
+++ b/LLama/Utils.cs
@@ -18,6 +18,7 @@ namespace LLama
lparams.n_ctx = @params.n_ctx;
lparams.n_parts = @params.n_parts;
+ lparams.n_gpu_layers = @params.n_gpu_layers;
lparams.seed = @params.seed;
lparams.f16_kv = @params.memory_f16;
lparams.use_mmap = @params.use_mmap;