Skip to content

Add model name to the progress bar

Bruno Cardoso requested to merge bc/add-model-name-to-progress into main

What does this merge request do and why?

Small change to include the model name to the BatchRequestCompletions progress bar. This is useful when multiple models are specified in the same run.

before:

BatchRequestCompletions: 4it [00:33,  7.96s/t]
BatchRequestCompletions: 4it [00:21,  5.36s/it]
BatchRequestCompletions: 4it [00:24,  5.69s/it]
BatchRequestCompletions: 4it [00:26,  6.07s/it]

after:

Requesting completions from gpt-3.5-turbo-1106: 4it [00:33,  7.96s/t]
Requesting completions from gpt-4-turbo: 4it [00:21,  5.36s/it]
Requesting completions from gpt-4: 4it [00:24,  5.69s/it]
Requesting completions from gpt-3.5-turbo: 4it [00:26,  6.07s/it]

How to set up and validate locally

Numbered steps to set up and validate the change are strongly suggested.

Merge request checklist

  • I've ran the affected pipeline(s) to validate that nothing is broken.
  • Tests added for new functionality. If not, please raise an issue to follow up.
  • Documentation added/updated, if needed.
Edited by Bruno Cardoso

Merge request reports