From 6418d1a906f15ca5f96eaa7c07c795766a17f8ee Mon Sep 17 00:00:00 2001
From: Ivan Lee <84584280+ivanleekk@users.noreply.github.com>
Date: Sun, 17 Aug 2025 21:18:30 +0800
Subject: [PATCH] fix: undo change and add 2 gpu

---
 main.slurm                           | 2 +-
 tradingagents/graph/trading_graph.py | 6 +++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/main.slurm b/main.slurm
index 9cf325db..2d348927 100644
--- a/main.slurm
+++ b/main.slurm
@@ -6,7 +6,7 @@
 #SBATCH --partition=gpu
 #SBATCH --ntasks=1
 #SBATCH --cpus-per-task=4
-#SBATCH --gpus=h100-96:1
+#SBATCH --gpus=h100-96:2
 #SBATCH --mem=96G
 
 # Run the script
diff --git a/tradingagents/graph/trading_graph.py b/tradingagents/graph/trading_graph.py
index 0fe3239c..fe9b9f6c 100644
--- a/tradingagents/graph/trading_graph.py
+++ b/tradingagents/graph/trading_graph.py
@@ -96,7 +96,11 @@ class TradingAgentsGraph:
                 trust_remote_code=True,
                 tensor_parallel_size=1,
             )
-            self.quick_thinking_llm = self.deep_thinking_llm
+            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']}")