fix build errors

This commit is contained in:
Bruce MacDonald
2023-08-09 10:45:57 -04:00
parent 7a5f3616fd
commit 09d8bf6730
2 changed files with 3 additions and 3 deletions

View File

@@ -416,7 +416,7 @@ func (llm *LLM) Embedding(input string) ([]float64, error) {
return nil, errors.New("llama: embedding not enabled")
}
tokens := llm.tokenize(input)
tokens := llm.Encode(input)
if tokens == nil {
return nil, errors.New("llama: tokenize embedding")
}