From 4f7b6ffdcc3e1aa3af0fc955726764271d92e0cf Mon Sep 17 00:00:00 2001 From: Martin Evans Date: Sat, 9 Sep 2023 01:09:27 +0100 Subject: [PATCH] Removed `GenerateResult` method that was only used in one place --- LLama/LLamaContext.cs | 6 ------ LLama/LLamaExecutorBase.cs | 6 ++---- 2 files changed, 2 insertions(+), 10 deletions(-) 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);