1 files changed, 1 insertions, 0 deletions
diff --git a/common/train.h b/common/train.h
index 42fa704b..d86c93cc 100644
--- a/common/train.h
+++ b/common/train.h
@@ -44,6 +44,7 @@ struct train_params_common {
     int n_batch;
     int n_gradient_accumulation;
     int n_epochs;
+    int n_gpu_layers;
 
     bool custom_n_ctx;