VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Regenerate failed file

Browse Source 
If a latent file fails to load, print out the path and the error, then return false to regenerate it
This commit is contained in:
Cauldrath
2024-04-21 13:46:31 -04:00
parent feefcf256e
commit 040e26ff1d
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
library/train_util.py
View File

@@ -2135,8 +2135,10 @@ def is_disk_cached_latents_is_expected(reso, npz_path: str, flip_aug: bool):
return False return False
if npz["latents_flipped"].shape[1:3] != expected_latents_size: if npz["latents_flipped"].shape[1:3] != expected_latents_size:
return False return False
except: except Exception as e:
raise RuntimeError(f"Error loading file: {npz_path}") print(npz_path)
print(e)
return False
return True return True