Skip to content
This repository has been archived by the owner on Aug 30, 2024. It is now read-only.

Commit

Permalink
fix format error
Browse files Browse the repository at this point in the history
Signed-off-by: intellinjun <[email protected]>
  • Loading branch information
intellinjun committed Mar 21, 2024
1 parent 3106b1b commit 3c6fc12
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion neural_speed/application/main_pybind.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -925,7 +925,7 @@ PYBIND11_MODULE(mixtral_cpp, m)

PYBIND11_MODULE(chatglm3_cpp, m)
#elif MODEL_NAME_ID == 21

PYBIND11_MODULE(gemma_cpp, m)

#endif
Expand Down
2 changes: 1 addition & 1 deletion neural_speed/application/main_run.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -229,7 +229,7 @@ int main(int argc, char** argv) { // NOLINT

// tokenize the prompt
bool add_bos = false;
if (params.model_arch == MODEL_LLAMA||params.model_arch == MODEL_GEMMA) {
if (params.model_arch == MODEL_LLAMA || params.model_arch == MODEL_GEMMA) {
add_bos = true;
}

Expand Down

0 comments on commit 3c6fc12

Please sign in to comment.