Fix RWKV tokenizer message

This commit is contained in:
somebody
2022-10-23 13:41:14 -05:00
parent 6f7e592f5e
commit d459d2a65e

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!")
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
from RWKV4.src.model_run import RWKV_RNN