Update examples/embedding/embedding.cpp
Co-authored-by: Xuan Son Nguyen <thichthat@gmail.com>
This commit is contained in:
parent
6e27406352
commit
d0541094a1
1 changed files with 1 additions and 1 deletions
|
@ -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();
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue