Skip to content

Commit 3701507

Browse files
committed
raise original error if error is occured in checking latents
1 parent 7802093 commit 3701507

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

library/train_util.py

+2-3
Original file line numberDiff line numberDiff line change
@@ -2136,9 +2136,8 @@ def is_disk_cached_latents_is_expected(reso, npz_path: str, flip_aug: bool):
21362136
if npz["latents_flipped"].shape[1:3] != expected_latents_size:
21372137
return False
21382138
except Exception as e:
2139-
print(npz_path)
2140-
print(e)
2141-
return False
2139+
logger.error(f"Error loading file: {npz_path}")
2140+
raise e
21422141

21432142
return True
21442143

0 commit comments

Comments
 (0)