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")
|