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']}")