Update examples/embedding/embedding.cpp

Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
This commit is contained in:
Minsoo Cheong 2024-03-26 09:25:16 +09:00 committed by GitHub
parent 6e27406352
commit d0541094a1
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -61,7 +61,7 @@ int main(int argc, char ** argv) {
} }
params.embedding = true; 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; params.n_ubatch = params.n_batch;
print_build_info(); print_build_info();