diff --git a/load_image_sequence.py b/load_image_sequence.py index 135381c..6a47cd6 100644 --- a/load_image_sequence.py +++ b/load_image_sequence.py @@ -62,9 +62,15 @@ class BatchLoadImageSequence: raise RuntimeError("Image sequence empty - no images to load") from concurrent.futures import ThreadPoolExecutor + from comfy.utils import ProgressBar + + pbar = ProgressBar(len(image_paths)) + imgs = [] with ThreadPoolExecutor() as executor: - imgs = list(executor.map(load_image, image_paths)) + for img in executor.map(load_image, image_paths): + imgs.append(img) + pbar.update(1) return (torch.cat(imgs, dim=0),)