update to rwkv 6 world

app.py

@@ -5,9 +5,10 @@ from rwkv.utils import PIPELINE, PIPELINE_ARGS
 from huggingface_hub import hf_hub_download
 
 ctx_limit = 4096
-title = "RWKV-5-World-1B5-v2-20231025-ctx4096"
-
-model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{title}.pth")
+# title = "RWKV-5-World-1B5-v2-20231025-ctx4096"
+# "BlinkDL/rwkv-5-world"
+title = "RWKV-x060-World-1B6-v2-20240208-ctx4096.pth"
+model_path = hf_hub_download(repo_id="BlinkDL/rwkv-6-world", filename=f"{title}.pth")
 model = RWKV(model=model_path, strategy="cpu bf16")
 pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
 
@@ -25,8 +26,8 @@ def generate_prompt(instruction, input=None, history=None):
         for pair in history:
             if pair[0] is not None and pair[1] is not None and len(pair[1]) > 0:
                 input += f"{pair[0]},{pair[1]},"
-        input += f"{instruction}"
-        instruction = "Generate a pirate Response using the following information."
+        input = input[-1] + f". {instruction}"
+        instruction = "Generate a Response using the following query."
     if input and len(input) > 0:
         input = (
             input.strip()