浏览代码

use strings.TrimLeft to remove spaces

Arne Müller 1 年之前
父节点
当前提交
46b9953f32
共有 1 个文件被更改,包括 2 次插入4 次删除
  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
 		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
 	var nextContext strings.Builder
 	nextContext.WriteString(prevConvo)
 	nextContext.WriteString(prevConvo)