Hugging Face's logo

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

App Files Community
10
BlinkDL commited on
Commit
c556c98
·
verified ·
1 Parent(s): 207190d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -24,9 +24,8 @@ gen_limit = 1000
24
  ########################## text rwkv ################################################################
25
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
26
 
27
- title = "rwkv7-g0a-7.2b-20250829-ctx4096"
28
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
29
- # model_path = "/home/molly/rwkv7-g0a-7.2b-20250829-ctx4096.pth"
30
  model = RWKV(model=model_path.replace('.pth',''), strategy='cuda fp16')
31
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
32
 
 
24
  ########################## text rwkv ################################################################
25
  from rwkv.utils import PIPELINE, PIPELINE_ARGS
26
 
27
+ title = "rwkv7-g0a2-7.2b-20251005-ctx4096"
28
  model_path = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
 
29
  model = RWKV(model=model_path.replace('.pth',''), strategy='cuda fp16')
30
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
31