Fix missing model key in bulk validate results on error

This commit is contained in:
Ross Wightman 2025-01-28 13:20:40 -08:00
parent b3a83b81d6
commit b1752eefb5

View File

@ -412,6 +412,7 @@ def _try_run(args, initial_batch_size):
break
batch_size = decay_batch_step(batch_size)
_logger.warning(f'Reducing batch size to {batch_size} for retry.')
results['model'] = args.model
results['error'] = error_str
_logger.error(f'{args.model} failed to validate ({error_str}).')
return results