fix: undo change and add 2 gpu
This commit is contained in:
parent
c300cc2819
commit
6418d1a906
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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']}")
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue