diff --git a/nanocode.py b/nanocode.py index f670616..d29b248 100644 --- a/nanocode.py +++ b/nanocode.py @@ -328,7 +328,7 @@ class RLMCodingProgram(PrecompiledProgram): self.agent = agent print(f"Using model: {self.agent.get_lm().model}") - print(f"Using sub-model: {self.agent.get_lm().model}") + print(f"Using sub-model: {self.agent.sub_lm.model}") def forward(self, task: str) -> str: assert task, "Task cannot be empty"