瀏覽代碼

use TrimPrefix instead of TrimLeft

Arne Müller 1 年之前
父節點
當前提交
730996e530
共有 1 個文件被更改,包括 1 次插入1 次删除
  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)