Update app.py
Browse files
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}
|
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 |
|