diff --git a/examples/csharp/HelloPhi/Program.cs b/examples/csharp/HelloPhi/Program.cs index fe0895744b..c23ab9907c 100644 --- a/examples/csharp/HelloPhi/Program.cs +++ b/examples/csharp/HelloPhi/Program.cs @@ -155,9 +155,13 @@ static string GetPrompt(bool interactive) using var generator = new Generator(model, generatorParams); generator.AppendTokenSequences(sequences); var watch = System.Diagnostics.Stopwatch.StartNew(); - while (!generator.IsDone()) + while (true) { generator.GenerateNextToken(); + if (generator.IsDone()) + { + break; + } Console.Write(tokenizerStream.Decode(generator.GetNextTokens()[0])); } Console.WriteLine(); @@ -192,9 +196,13 @@ static string GetPrompt(bool interactive) var sequences = tokenizer.Encode(tokenizer.ApplyChatTemplate("", messages, "", true)); var watch = System.Diagnostics.Stopwatch.StartNew(); generator.AppendTokenSequences(sequences); - while (!generator.IsDone()) + while (true) { generator.GenerateNextToken(); + if (generator.IsDone()) + { + break; + } Console.Write(tokenizerStream.Decode(generator.GetNextTokens()[0])); } Console.WriteLine();