diff options
Diffstat (limited to 'examples/train-text-from-scratch/train-text-from-scratch.cpp')
| -rw-r--r-- | examples/train-text-from-scratch/train-text-from-scratch.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/train-text-from-scratch/train-text-from-scratch.cpp b/examples/train-text-from-scratch/train-text-from-scratch.cpp index 05bfa80..c50eeb3 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -2671,7 +2671,8 @@ struct train_params { const char * fn_checkpoint_out; const char * fn_model_out; - int seed; + uint32_t seed; + int n_ctx; int n_embd; int n_mult; |