diff --git a/projects/singularity_cinema/generate_script/agent.py b/projects/singularity_cinema/generate_script/agent.py index 0586aafc2..0805dd931 100644 --- a/projects/singularity_cinema/generate_script/agent.py +++ b/projects/singularity_cinema/generate_script/agent.py @@ -31,7 +31,7 @@ def prepare_llm(self): config = deepcopy(self.config) config.generation_config.temperature = 0.6 config.generation_config.top_k = 50 - self.llm: LLM = LLM.from_config(self.config) + self.llm: LLM = LLM.from_config(config) def on_task_end(self, messages: List[Message]): script = os.path.join(self.work_dir, 'script.txt')