Hugging Face's logo

Spaces:
BlinkDL
/
RWKV-Gradio-1
Running on T4

App Files Community
4
BlinkDL commited on
Commit
1203d1a
·
verified ·
1 Parent(s): b233fba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ ctx_limit = 3500
22
  ########################## text rwkv ################################################################
23
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
24
 
25
- title_v6 = "RWKV-x060-World-1B6-v2-20240208-ctx4096"
26
  model_path_v6 = hf_hub_download(repo_id="BlinkDL/rwkv-6-world", filename=f"{title_v6}.pth")
27
  model_v6 = RWKV(model=model_path_v6, strategy='cuda fp16')
28
  pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")
 
22
  ########################## text rwkv ################################################################
23
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
24
 
25
+ title_v6 = "RWKV-x060-World-3B-v2-20240228-ctx4096"
26
  model_path_v6 = hf_hub_download(repo_id="BlinkDL/rwkv-6-world", filename=f"{title_v6}.pth")
27
  model_v6 = RWKV(model=model_path_v6, strategy='cuda fp16')
28
  pipeline_v6 = PIPELINE(model_v6, "rwkv_vocab_v20230424")