raise original error if error is occured in checking latents

feffy380 · May 12, 2024 · 3701507 · 3701507
1 parent 7802093
commit 3701507
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -2136,9 +2136,8 @@ def is_disk_cached_latents_is_expected(reso, npz_path: str, flip_aug: bool):
             if npz["latents_flipped"].shape[1:3] != expected_latents_size:
                 return False
     except Exception as e:
-        print(npz_path)
-        print(e)
-        return False
+        logger.error(f"Error loading file: {npz_path}")
+        raise e
 
     return True