charieswei
/
TradingAgents
mirror of https://github.com/TauricResearch/TradingAgents.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix: undo change and add 2 gpu

This commit is contained in:
Ivan Lee 2025-08-17 21:18:30 +08:00
parent c300cc2819
commit 6418d1a906
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
main.slurm
View File

@ -6,7 +6,7 @@
#SBATCH --partition=gpu #SBATCH --partition=gpu
#SBATCH --ntasks=1 #SBATCH --ntasks=1
#SBATCH --cpus-per-task=4 #SBATCH --cpus-per-task=4
#SBATCH --gpus=h100-96:1 #SBATCH --gpus=h100-96:2
#SBATCH --mem=96G #SBATCH --mem=96G
# Run the script # Run the script

6
tradingagents/graph/trading_graph.py
View File

@ -96,7 +96,11 @@ class TradingAgentsGraph:
trust_remote_code=True, trust_remote_code=True,
tensor_parallel_size=1, 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: else:
raise ValueError(f"Unsupported LLM provider: {self.config['llm_provider']}") raise ValueError(f"Unsupported LLM provider: {self.config['llm_provider']}")