Ver Fonte

trim whitespace before checking stop conditions

Fixes #295
Jeffrey Morgan há 1 ano atrás
pai
commit
5eb712f962
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      llama/llama.go

+ 2 - 2
llama/llama.go

@@ -250,9 +250,9 @@ func (llm *LLM) Predict(ctx []int, prompt string, fn func(api.GenerateResponse))
 
 
 func (llm *LLM) checkStopConditions(b bytes.Buffer) error {
 func (llm *LLM) checkStopConditions(b bytes.Buffer) error {
 	for _, stopCondition := range llm.Stop {
 	for _, stopCondition := range llm.Stop {
-		if stopCondition == b.String() {
+		if stopCondition == strings.TrimSpace(b.String()) {
 			return io.EOF
 			return io.EOF
-		} else if strings.HasPrefix(stopCondition, b.String()) {
+		} else if strings.HasPrefix(stopCondition, strings.TrimSpace(b.String())) {
 			return errNeedMoreData
 			return errNeedMoreData
 		}
 		}
 	}
 	}