Use tokenizer in Agent
Browse files
agents.py
CHANGED
|
@@ -165,9 +165,11 @@ def build_graph(provider: str = "rwkv"):
|
|
| 165 |
title = "rwkv7-g1-0.1b-20250307-ctx4096"
|
| 166 |
pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
|
| 167 |
model_path = pth.replace(".pth", "")
|
|
|
|
| 168 |
raw_llm = RWKV(
|
| 169 |
model=model_path,
|
| 170 |
-
strategy="cpu fp32",
|
|
|
|
| 171 |
)
|
| 172 |
|
| 173 |
class RWKVWithTools:
|
|
@@ -209,6 +211,7 @@ def build_graph(provider: str = "rwkv"):
|
|
| 209 |
# --- END RWKV SETUP ---
|
| 210 |
else:
|
| 211 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
|
|
|
| 212 |
# Bind tools to LLM
|
| 213 |
llm_with_tools = llm.bind_tools(tools)
|
| 214 |
|
|
|
|
| 165 |
title = "rwkv7-g1-0.1b-20250307-ctx4096"
|
| 166 |
pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
|
| 167 |
model_path = pth.replace(".pth", "")
|
| 168 |
+
|
| 169 |
raw_llm = RWKV(
|
| 170 |
model=model_path,
|
| 171 |
+
strategy="cpu fp32",
|
| 172 |
+
tokens_path="./rwkv_vocab_v20230424.txt",
|
| 173 |
)
|
| 174 |
|
| 175 |
class RWKVWithTools:
|
|
|
|
| 211 |
# --- END RWKV SETUP ---
|
| 212 |
else:
|
| 213 |
raise ValueError("Invalid provider. Choose 'google', 'groq' or 'huggingface'.")
|
| 214 |
+
|
| 215 |
# Bind tools to LLM
|
| 216 |
llm_with_tools = llm.bind_tools(tools)
|
| 217 |
|