Browse Source

use strings.TrimLeft to remove spaces

Arne Müller 1 year ago
parent
commit
46b9953f32
1 changed files with 2 additions and 4 deletions
  1. 2 4
      llm/llama.go

+ 2 - 4
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)