Commit
•
5141f22
1
Parent(s):
6b2c851
fix typo
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ from huggingface_hub import hf_hub_download
|
|
| 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-20240208-ctx4096
|
| 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")
|
|
|
|
| 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-20240208-ctx4096"
|
| 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")
|