Update README.md

README.md

@@ -27,8 +27,8 @@ import gradio as gr
 os.environ["RWKV_JIT_ON"] = '1'
 os.environ["RWKV_CUDA_ON"] = '1'
 from rwkv.model import RWKV
-model_path = "chatntq-7b-jpntuned/ChatNTQ-7B-RWKV-world-JPNtuned-ctx2048.pth"
-WORD_NAME = "rwkv_vocab_v20230424" # copy rwkv_vocab_v20230424.txt in ChatNTQ-7B-Japanese to the same folder test
+model_path = "chatntq-7b-jpntuned/ChatNTQ-7B-RWKV-4-World-JPNtuned-ctx2048.pth"
+WORD_NAME = "rwkv_vocab_v20230424" # copy rwkv_vocab_v20230424.txt in chatntq-7b-jpntuned to the same folder test
 ctx_limit = 1024
 model = RWKV(model=model_path, strategy='cuda fp16i8 *24 -> cuda fp16')
 from rwkv.utils import PIPELINE, PIPELINE_ARGS