From d0541094a1b93123a5b6706503a809977d2da0d1 Mon Sep 17 00:00:00 2001 From: Minsoo Cheong <54794500+mscheong01@users.noreply.github.com> Date: Tue, 26 Mar 2024 09:25:16 +0900 Subject: [PATCH] Update examples/embedding/embedding.cpp Co-authored-by: Xuan Son Nguyen --- examples/embedding/embedding.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/embedding/embedding.cpp b/examples/embedding/embedding.cpp index f2dc38a81..9aede7fad 100644 --- a/examples/embedding/embedding.cpp +++ b/examples/embedding/embedding.cpp @@ -61,7 +61,7 @@ int main(int argc, char ** argv) { } params.embedding = true; - // For BERT models, batch size must be equal to ubatch size + // For non-causal models, batch size must be equal to ubatch size params.n_ubatch = params.n_batch; print_build_info();