diff --git a/LLama/LLamaContext.cs b/LLama/LLamaContext.cs index 1a1845b0..d2fb0fe3 100644 --- a/LLama/LLamaContext.cs +++ b/LLama/LLamaContext.cs @@ -482,12 +482,6 @@ namespace LLama } #endregion - internal IEnumerable GenerateResult(IEnumerable ids) - { - foreach(var id in ids) - yield return _ctx.TokenToString(id, _encoding); - } - /// /// Convert a token into a string /// diff --git a/LLama/LLamaExecutorBase.cs b/LLama/LLamaExecutorBase.cs index 0eefa98e..49bd3170 100644 --- a/LLama/LLamaExecutorBase.cs +++ b/LLama/LLamaExecutorBase.cs @@ -288,10 +288,8 @@ namespace LLama if (args.ReturnValue) { - foreach (var item in Context.GenerateResult(_embeds)) - { - yield return item; - } + foreach (var id in _embeds) + yield return Context.TokenToString(id); } var breakGeneration = PostProcess(inferenceParams, args, out var extraOutputs);