From 5ac6f1857bdab95196f8686f28eaba20b47e54c8 Mon Sep 17 00:00:00 2001
From: Ivan Lee <84584280+ivanleekk@users.noreply.github.com>
Date: Sun, 17 Aug 2025 20:57:13 +0800
Subject: [PATCH] fix: tensor_parallel_size=1

---
 tradingagents/graph/trading_graph.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py
index 83c2c4bf..fe9b9f6c 100644
--- a/tradingagents/graph/trading_graph.py
+++ b/tradingagents/graph/trading_graph.py
@@ -94,10 +94,12 @@ class TradingAgentsGraph:
             self.deep_thinking_llm = VLLM(
                 model=self.config["deep_think_llm"],
                 trust_remote_code=True,
+                tensor_parallel_size=1,
             )
             self.quick_thinking_llm = VLLM(
                 model=self.config["quick_think_llm"],
                 trust_remote_code=True,
+                tensor_parallel_size=1,
             )
         else:
             raise ValueError(f"Unsupported LLM provider: {self.config['llm_provider']}")