budivoy commited on
Commit
daf34ea
·
verified ·
1 Parent(s): 322605d

Update model name

Browse files
Files changed (1) hide show
  1. agents.py +1 -1
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")