Regenerate failed file

If a latent file fails to load, print out the path and the error, then return false to regenerate it
feffy380 · Apr 21, 2024 · 040e26f · 040e26f
1 parent feefcf2
commit 040e26f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/library/train_util.py b/library/train_util.py
@@ -2135,8 +2135,10 @@ def is_disk_cached_latents_is_expected(reso, npz_path: str, flip_aug: bool):
                 return False
             if npz["latents_flipped"].shape[1:3] != expected_latents_size:
                 return False
-    except:
-        raise RuntimeError(f"Error loading file: {npz_path}")
+    except Exception as e:
+        print(npz_path)
+        print(e)
+        return False
 
     return True