baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

use strings.TrimLeft to remove spaces

Browse source
This commit is contained in:
Arne Müller 2023-10-18 22:41:19 +02:00
parent 90c49bed57
commit 46b9953f32
1 changed files with 2 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
llm/llama.go
View file

@ -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)