hotfix and fonud unsolved issue

This commit is contained in:
Darren Laurie
2025-03-06 00:14:38 +08:00
parent 571fa8072a
commit 57cc9ea0ad
2 changed files with 2 additions and 0 deletions

View File

@@ -522,6 +522,7 @@ class LatentsCachingStrategy:
"""
return self._default_load_latents_from_disk(None, npz_path, bucket_reso)
# TODO: Solve Multinode hang
def _default_load_latents_from_disk(
self, latents_stride: Optional[int], npz_path: str, bucket_reso: Tuple[int, int]
) -> Tuple[Optional[np.ndarray], Optional[List[int]], Optional[List[int]], Optional[np.ndarray], Optional[np.ndarray]]:

View File

@@ -135,6 +135,7 @@ IMAGE_TRANSFORMS = transforms.Compose(
TEXT_ENCODER_OUTPUTS_CACHE_SUFFIX = "_te_outputs.npz"
TEXT_ENCODER_OUTPUTS_CACHE_SUFFIX_SD3 = "_sd3_te.npz"
SKIP_NPZ_PATH_CHECK = False
def set_skip_npz_path_check(skip: bool):
global SKIP_NPZ_PATH_CHECK