diff options
author | SIGSEGV <21287366+akr2002@users.noreply.github.com> | 2023-07-12 19:18:43 +0530 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-12 19:18:43 +0530 |
commit | 2516af4cd61f509c995b4f78fdf123cba33f3509 (patch) | |
tree | de7324f01b9454fb30e4d827b8300d02fd982ed3 /examples/train-text-from-scratch/train-text-from-scratch.cpp | |
parent | ff34a7d385fc47c4d432fd8c19306d5aca814d05 (diff) | |
parent | 4e7464ef88885cb3532738b03cac890f4077fa20 (diff) |
Merge branch 'ggerganov:master' into master
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 | 14 |
1 files changed, 3 insertions, 11 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 b96fdcd..afbb4a7 100644 --- a/examples/train-text-from-scratch/train-text-from-scratch.cpp +++ b/examples/train-text-from-scratch/train-text-from-scratch.cpp @@ -1354,17 +1354,9 @@ struct ggml_tensor * expand(struct ggml_cgraph * g, struct ggml_tensor * t) { } } - if (t->src0) { - expand(g, t->src0); - } - - if (t->src1) { - expand(g, t->src1); - } - - for (int i = 0; i < GGML_MAX_OPT; ++i) { - if (t->opt[i]) { - expand(g, t->opt[i]); + for (int i = 0; i < GGML_MAX_SRC; ++i) { + if (t->src[i]) { + expand(g, t->src[i]); } } |