validations fix for deepspeed

bghira · Oct 10, 2024 · 51d9f13 · 51d9f13
1 parent 1942d5b
commit 51d9f13
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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