From 46b9953f32d665010367f55362aff41b4b338a8d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Arne=20M=C3=BCller?= Date: Wed, 18 Oct 2023 22:41:19 +0200 Subject: [PATCH] use strings.TrimLeft to remove spaces --- llm/llama.go | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/llm/llama.go b/llm/llama.go index 97685645..bd5c3bb7 100644 --- a/llm/llama.go +++ b/llm/llama.go @@ -464,10 +464,8 @@ func (llm *llama) Predict(ctx context.Context, prevContext []int, prompt string, return err } - // Remove first leading space from prevConvo if present - if len(prevConvo) > 0 && prevConvo[0] == ' ' { - prevConvo = prevConvo[1:] - } + // Remove leading spaces from prevConvo if present + prevConvo = strings.TrimLeft(prevConvo, " ") var nextContext strings.Builder nextContext.WriteString(prevConvo)