FredZhang7 commited on
Commit
62c6c0f
·
1 Parent(s): dd4df2b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from rwkv.model import RWKV
5
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
6
 
7
  ctx_limit = 2048
8
- title = "RWKV-5-World-1B5-v2-20231025-ctx4096.pth"
9
 
10
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{title}")
11
  model = RWKV(model=model_path, strategy='cpu bf16')
@@ -102,7 +102,7 @@ def alternative(chatbot, history):
102
  with gr.Blocks(title=title) as demo:
103
  gr.HTML(f"<div style=\"text-align: center;\">\n<h1>🌍World - {title}</h1>\n</div>")
104
  with gr.Tab("Instruct mode"):
105
- gr.Markdown(f"World is RWKV-5 World 1.5B 100% RNN RWKV-LM **trained on 100+ world languages**. Demo limited to ctxlen {ctx_limit}. Finetuned on alpaca, gpt4all, codealpaca and more. For best results, ** keep you prompt short and clear **.</b>.") # <b>UPDATE: now with Chat (see above, as a tab) ==> turn off as of now due to VRAM leak caused by buggy code.
106
  with gr.Row():
107
  with gr.Column():
108
  instruction = gr.Textbox(lines=2, label="Instruction", value='東京で訪れるべき素晴らしい場所とその紹介をいくつか挙げてください。')
 
5
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
6
 
7
  ctx_limit = 2048
8
+ title = "RWKV-5-World-0.1B-v1-20230803-ctx4096.pth"
9
 
10
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv-5-world", filename=f"{title}")
11
  model = RWKV(model=model_path, strategy='cpu bf16')
 
102
  with gr.Blocks(title=title) as demo:
103
  gr.HTML(f"<div style=\"text-align: center;\">\n<h1>🌍World - {title}</h1>\n</div>")
104
  with gr.Tab("Instruct mode"):
105
+ gr.Markdown(f"100% RNN RWKV-LM **trained on 100+ world languages**. Demo limited to ctxlen {ctx_limit}. Finetuned on alpaca, gpt4all, codealpaca and more. For best results, ** keep you prompt short and clear **.</b>.") # <b>UPDATE: now with Chat (see above, as a tab) ==> turn off as of now due to VRAM leak caused by buggy code.
106
  with gr.Row():
107
  with gr.Column():
108
  instruction = gr.Textbox(lines=2, label="Instruction", value='東京で訪れるべき素晴らしい場所とその紹介をいくつか挙げてください。')