From b1752eefb513c7828c302b8873b50a0ed7049409 Mon Sep 17 00:00:00 2001
From: Ross Wightman <rwightman@gmail.com>
Date: Tue, 28 Jan 2025 13:20:40 -0800
Subject: [PATCH] Fix missing model key in bulk validate results on error

---
 validate.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/validate.py b/validate.py
index 2e416e65..f757855e 100755
--- a/validate.py
+++ b/validate.py
@@ -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