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

Commit

Permalink
remove cout
Browse files Browse the repository at this point in the history
  • Loading branch information
Zhenzhong1 authored Mar 21, 2024
1 parent 6729a64 commit dcfe522
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions neural_speed/application/main_run.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -238,8 +238,6 @@ int main(int argc, char** argv) { // NOLINT
std::vector<std::string> prompts;
prompts.push_back(params.prompt);
std::string prompt = build_prompt_glm2(prompts);
std::cout << "prompt = " << prompt << std::endl;

embd_inp = ::model_tokenize(ctx, prompt, false);
embd_inp.insert(embd_inp.begin(), {64790, 64792}); // special prefix
} else if (params.model_arch == MODEL_CHATGLM || params.model_arch == MODEL_BAICHUAN ||
Expand Down

0 comments on commit dcfe522

Please sign in to comment.