rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Fix RWKV tokenizer message

Browse Source
This commit is contained in:
somebody
2022-10-23 13:41:14 -05:00
parent 6f7e592f5e
commit d459d2a65e
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
aiserver.py
View File

@@ -5848,7 +5848,10 @@ def rwkv_init(model_class: str, use_gpu: bool = False):
raise RuntimeError(f"No config for model '{model_class}' found!") raise RuntimeError(f"No config for model '{model_class}' found!")
if not os.path.exists(TOKENIZER_PATH): if not os.path.exists(TOKENIZER_PATH):
raise RuntimeError(f"Can't find tokenizer at '{TOKENIZER_PATH}'!") raise RuntimeError(
f"Can't find tokenizer at '{TOKENIZER_PATH}'! Please download it from " \
f"https://raw.githubusercontent.com/BlinkDL/RWKV-LM/main/RWKV-v4/20B_tokenizer.json and place it at '{TOKENIZER_PATH}"
)
# Model stuff # Model stuff
from RWKV4.src.model_run import RWKV_RNN from RWKV4.src.model_run import RWKV_RNN