[API] Fix wrong file for nerdstash v2

Aedial · May 20, 2023 · 80797d0 · 80797d0
1 parent 38dbf93
commit 80797d0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/novelai_api/Tokenizer.py b/novelai_api/Tokenizer.py
@@ -60,7 +60,7 @@ def get_tokenizer_name(cls, model: Model) -> str:
     _NERDSTASH_TOKENIZER_v1.decode = _NERDSTASH_TOKENIZER_v1.DecodeIds
 
     _NERDSTASH_TOKENIZER_v2 = sentencepiece.SentencePieceProcessor()
-    _NERDSTASH_TOKENIZER_v2.Load(str(tokenizers_path / "nerdstash_v1.model"))
+    _NERDSTASH_TOKENIZER_v2.Load(str(tokenizers_path / "nerdstash_v2.model"))
     _NERDSTASH_TOKENIZER_v2.encode = _NERDSTASH_TOKENIZER_v2.EncodeAsIds
     _NERDSTASH_TOKENIZER_v2.decode = _NERDSTASH_TOKENIZER_v2.DecodeIds