From 51d9f13ebecec4a3aab27be0361c756bd2c9e466 Mon Sep 17 00:00:00 2001 From: bghira Date: Wed, 9 Oct 2024 19:20:18 -0600 Subject: [PATCH] validations fix for deepspeed --- helpers/training/validation.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/helpers/training/validation.py b/helpers/training/validation.py index ac0344ce..f40127af 100644 --- a/helpers/training/validation.py +++ b/helpers/training/validation.py @@ -837,6 +837,8 @@ def save_benchmark(self, benchmark: str = "base_model"): base_model_benchmark = self._benchmark_path(benchmark=benchmark) if not os.path.exists(base_model_benchmark): os.makedirs(base_model_benchmark, exist_ok=True) + if self.validation_images is None: + return for shortname, image_list in self.validation_images.items(): for idx, image in enumerate(image_list): width, height = image.size