diff options
Diffstat (limited to 'examples/embedding')
| -rw-r--r-- | examples/embedding/embedding.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/examples/embedding/embedding.cpp b/examples/embedding/embedding.cpp index a553ae1c..49302a19 100644 --- a/examples/embedding/embedding.cpp +++ b/examples/embedding/embedding.cpp @@ -107,7 +107,7 @@ int main(int argc, char ** argv) { // max batch size const uint64_t n_batch = params.n_batch; - GGML_ASSERT(params.n_batch == params.n_ctx); + GGML_ASSERT(params.n_batch >= params.n_ctx); // tokenize the prompts and trim std::vector<std::vector<int32_t>> inputs; |