diff options
Diffstat (limited to 'common/train.h')
| -rw-r--r-- | common/train.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/common/train.h b/common/train.h index d86c93cc..263d940c 100644 --- a/common/train.h +++ b/common/train.h @@ -9,6 +9,8 @@ #include "ggml.h" #include "llama.h" +#define LLAMA_TRAIN_MAX_NODES 16384 + typedef std::string mt19937_state; struct train_state { |