Shago commited on
Commit
57f969a
·
verified ·
1 Parent(s): 83a2313

Update agents/agents_nodes.py

Browse files
Files changed (1) hide show
  1. agents/agents_nodes.py +2 -2
agents/agents_nodes.py CHANGED
@@ -14,10 +14,10 @@ from tools.financial_tools import time_value_tool
14
 
15
  text_generator = pipeline(
16
  "text-generation", # Task type
17
- model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",
18
  # model="google/gemma-3n-E2B-it",
19
  # model="google/gemma-3n-e4b-it",
20
- # model="Qwen/Qwen3-Embedding-0.6B",
21
  # device="cuda" if torch.cuda.is_available() else "cpu",
22
  device= "cpu",
23
  torch_dtype=torch.bfloat16,
 
14
 
15
  text_generator = pipeline(
16
  "text-generation", # Task type
17
+ # model="deepseek-ai/DeepSeek-R1-0528-Qwen3-8B",
18
  # model="google/gemma-3n-E2B-it",
19
  # model="google/gemma-3n-e4b-it",
20
+ model="Qwen/Qwen3-Embedding-0.6B",
21
  # device="cuda" if torch.cuda.is_available() else "cpu",
22
  device= "cpu",
23
  torch_dtype=torch.bfloat16,