Update agents.py

agents.py

@@ -14,7 +14,7 @@ from langchain_community.vectorstores import SupabaseVectorStore
 from langchain_core.messages import SystemMessage, HumanMessage
 from langchain_core.tools import tool
 
-os.environ["RWKV_V7_ON"] = '0' # ==> enable RWKV-7 mode
+os.environ["RWKV_V7_ON"] = '1' # ==> enable RWKV-7 mode
 os.environ['RWKV_JIT_ON'] = '1' # '1' for better speed
 os.environ["RWKV_CUDA_ON"] = '0' # '1' to compile CUDA kernel (10x faster), requires c++ compiler & cuda libraries
 
@@ -172,7 +172,7 @@ def build_graph(provider: str = "rwkv"):
         pth = hf_hub_download(repo_id="BlinkDL/rwkv7-g1", filename=f"{title}.pth")
         model_path = pth.replace(".pth", "")
 
-        raw_llm = RWKV(model=model_path, strategy='cuda fp32')
+        raw_llm = RWKV(model=model_path, strategy='cpu fp32')
         pipeline = PIPELINE(raw_llm, "rwkv_vocab_v20230424")
 
         class RWKVWithTools: