FredZhang7 commited on
Commit
9119294
1 Parent(s): 8bd55c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ ctx_limit = 4096
8
  # title = "RWKV-5-World-1B5-v2-20231025-ctx4096"
9
  # "BlinkDL/rwkv-5-world"
10
  title = "RWKV-x060-World-1B6-v2.1-20240328-ctx4096.pth"
11
- model_path = hf_hub_download(repo_id="BlinkDL/rwkv-6-world", filename=f"{title}.pth")
12
  model = RWKV(model=model_path, strategy="cpu bf16")
13
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
14
 
 
8
  # title = "RWKV-5-World-1B5-v2-20231025-ctx4096"
9
  # "BlinkDL/rwkv-5-world"
10
  title = "RWKV-x060-World-1B6-v2.1-20240328-ctx4096.pth"
11
+ model_path = hf_hub_download(repo_id="BlinkDL/rwkv-6-world", filename=f"{title}")
12
  model = RWKV(model=model_path, strategy="cpu bf16")
13
  pipeline = PIPELINE(model, "rwkv_vocab_v20230424")
14