From 1da6c40f4f1335ad684bed447243227b20aef36e Mon Sep 17 00:00:00 2001 From: jmorganca Date: Thu, 13 Jun 2024 14:21:55 -0700 Subject: [PATCH] lint --- llama/llama.go | 2 +- llama/runner/runner.go | 5 ++++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/llama/llama.go b/llama/llama.go index 55501598..db2b522d 100644 --- a/llama/llama.go +++ b/llama/llama.go @@ -302,7 +302,7 @@ func (m *Model) Tokenize(text string, addSpecial bool, parseSpecial bool) ([]int } tokens := make([]int, result) - for i := 0; i < int(result); i++ { + for i := range result { tokens[i] = int(cTokens[i]) } diff --git a/llama/runner/runner.go b/llama/runner/runner.go index 6fad0d95..05f436ad 100644 --- a/llama/runner/runner.go +++ b/llama/runner/runner.go @@ -441,7 +441,10 @@ func main() { server.model = llama.LoadModelFromFile(*mpath, params) if *lpath != "" { - server.model.ApplyLoraFromFile(*lpath, 1.0, "", *threads) + err := server.model.ApplyLoraFromFile(*lpath, 1.0, "", *threads) + if err != nil { + panic(err) + } } ctxParams := llama.NewContextParams(*numCtx, *threads, *flashAttention)