diff options
Diffstat (limited to 'tokenizer/bpe/utility.go')
| -rw-r--r-- | tokenizer/bpe/utility.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tokenizer/bpe/utility.go b/tokenizer/bpe/utility.go index fbe99fc..f8fcdbc 100644 --- a/tokenizer/bpe/utility.go +++ b/tokenizer/bpe/utility.go @@ -2,7 +2,7 @@ package bpe import "github.com/jonasknobloch/mbpe" -func NewTokenizerFromFiles(vocab, merges string) (*Tokenizer, error) { +func NewTokenizerFromFiles(vocab, merges string, cfg Config) (*Tokenizer, error) { model := mbpe.NewMBPE() if err := model.Load(vocab, merges); err != nil { @@ -16,7 +16,7 @@ func NewTokenizerFromFiles(vocab, merges string) (*Tokenizer, error) { tokenizer.SetPreTokenizer(pre) tokenizer.SetDecoder(pre) - return NewTokenizer(tokenizer), nil + return NewTokenizer(tokenizer, cfg), nil } func Vocab(t *Tokenizer) []string { |