Browse Source

use TrimPrefix instead of TrimLeft

Arne Müller 1 year ago
parent
commit
730996e530
1 changed files with 1 additions and 1 deletions
  1. 1 1
      llm/llama.go

+ 1 - 1
llm/llama.go

@@ -465,7 +465,7 @@ func (llm *llama) Predict(ctx context.Context, prevContext []int, prompt string,
 	}
 
 	// Remove leading spaces from prevConvo if present
-	prevConvo = strings.TrimLeft(prevConvo, " ")
+	prevConvo = strings.TrimPrefix(prevConvo, " ")
 
 	var nextContext strings.Builder
 	nextContext.WriteString(prevConvo)