Update model name
Browse files
agents.py
CHANGED
|
@@ -172,7 +172,7 @@ def build_graph(provider: str = "rwkv"):
|
|
| 172 |
)
|
| 173 |
elif provider == "rwkv":
|
| 174 |
# --- BEGIN RWKV SETUP ---
|
| 175 |
-
title = "rwkv7-g1"
|
| 176 |
pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
|
| 177 |
# 2) Load RWKV (drop .pth extension for RWKV loader)
|
| 178 |
rwkv_model = RWKV(model=pth.replace(".pth", ""), strategy="cpu fp32")
|
|
|
|
| 172 |
)
|
| 173 |
elif provider == "rwkv":
|
| 174 |
# --- BEGIN RWKV SETUP ---
|
| 175 |
+
title = "rwkv7-g1-0.1b-20250307-ctx4096"
|
| 176 |
pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
|
| 177 |
# 2) Load RWKV (drop .pth extension for RWKV loader)
|
| 178 |
rwkv_model = RWKV(model=pth.replace(".pth", ""), strategy="cpu fp32")
|