Spaces:
Runtime error
Runtime error
FredZhang7
commited on
Commit
·
62c6c0f
1
Parent(s):
dd4df2b
Update app.py
Browse files
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-
|
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"
|
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='東京で訪れるべき素晴らしい場所とその紹介をいくつか挙げてください。')
|