|
|
|
@ -4,7 +4,7 @@ if __name__ == "__main__":
|
|
|
|
|
# Initialize the ReasoningDuo with two lightweight models
|
|
|
|
|
duo = ReasoningDuo(
|
|
|
|
|
model_names=["gpt-4o-mini", "gpt-4o-mini"],
|
|
|
|
|
max_loops=1,
|
|
|
|
|
# max_loops=1, # Remove this line
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Batched tasks to process
|
|
|
|
@ -16,4 +16,4 @@ if __name__ == "__main__":
|
|
|
|
|
# Run the batch once and print each result
|
|
|
|
|
results = duo.batched_run(tasks)
|
|
|
|
|
for task, output in zip(tasks, results):
|
|
|
|
|
print(f"Task: {task}\nResult: {output}\n")
|
|
|
|
|
print(f"Task: {task}\nResult: {output}\n")
|