debug
This commit is contained in:
@@ -323,11 +323,13 @@ class RLMCodingProgram(PrecompiledProgram):
|
|||||||
max_iterations=self.config.max_iters,
|
max_iterations=self.config.max_iters,
|
||||||
verbose=self.config.verbose,
|
verbose=self.config.verbose,
|
||||||
)
|
)
|
||||||
|
|
||||||
agent.set_lm(self.lm)
|
agent.set_lm(self.lm)
|
||||||
print(f"Using model: {self.lm.model}")
|
|
||||||
print(f"Using sub-model: {self.sub_lm.model}")
|
|
||||||
self.agent = agent
|
self.agent = agent
|
||||||
|
|
||||||
|
print(f"Using model: {self.agent.get_lm().model}")
|
||||||
|
print(f"Using sub-model: {self.agent.get_lm().model}")
|
||||||
|
|
||||||
def forward(self, task: str) -> str:
|
def forward(self, task: str) -> str:
|
||||||
assert task, "Task cannot be empty"
|
assert task, "Task cannot be empty"
|
||||||
return self.agent(task=task)
|
return self.agent(task=task)
|
||||||
|
|||||||
Reference in New Issue
Block a user