fix(rwkv): load tokenizer file from model path (#255)

2 years ago · 2a9d7474ce
parent 850a690290
commit 2a9d7474ce
1 changed files with 2 additions and 1 deletions
--- a/pkg/model/initializers.go
+++ b/pkg/model/initializers.go
@ -2,6 +2,7 @@ package model

 import (
 	"fmt"
+	"path/filepath"
 	"strings"

 	rwkv "github.com/donomii/go-rwkv.cpp"
@ -143,7 +144,7 @@ func (ml *ModelLoader) BackendLoader(backendString string, modelFile string, lla
 	case BertEmbeddingsBackend:
 		return ml.LoadModel(modelFile, bertEmbeddings)
 	case RwkvBackend:
-		return ml.LoadModel(modelFile, rwkvLM(modelFile+tokenizerSuffix, threads))
+		return ml.LoadModel(modelFile, rwkvLM(filepath.Join(ml.ModelPath, modelFile+tokenizerSuffix), threads))
 	case WhisperBackend:
 		return ml.LoadModel(modelFile, whisperModel)
 	default: