From e70e3c9e477802b802f113107b64322d8046dba0 Mon Sep 17 00:00:00 2001 From: Ryan Hill Date: Tue, 22 Oct 2024 01:21:00 -0700 Subject: [PATCH] Revert "Test debug" This reverts commit 261726743ce82c3aa827f78add86efd12cbcfcf8. --- test/c_api_tests.cpp | 10 ---------- 1 file changed, 10 deletions(-) diff --git a/test/c_api_tests.cpp b/test/c_api_tests.cpp index 9b85a1769..c04ef284f 100644 --- a/test/c_api_tests.cpp +++ b/test/c_api_tests.cpp @@ -307,19 +307,13 @@ struct Phi2Test { void Run() { // Low level loop { - std::cout << "Creating generator..." << std::endl; - auto generator = OgaGenerator::Create(*model_, *params_); - std::cout << "Generation loop..." << std::endl; - while (!generator->IsDone()) { generator->ComputeLogits(); generator->GenerateNextToken(); } - std::cout << "Decode one at a time" << std::endl; - // Decode One at a time for (size_t i = 0; i < 3; i++) { auto out_string = tokenizer_->Decode(generator->GetSequenceData(i), generator->GetSequenceCount(i)); @@ -329,12 +323,8 @@ struct Phi2Test { // High level { - std::cout << "High level generate" << std::endl; - auto output_sequences = model_->Generate(*params_); - std::cout << "Decode the batch" << std::endl; - // Decode The Batch for (size_t i = 0; i < output_sequences->Count(); i++) { auto out_string = tokenizer_->Decode(output_sequences->SequenceData(i), output_sequences->SequenceCount(i));