Update agents/agents_nodes.py
Browse files- 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 |
-
|
| 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,
|