diff --git a/iz_helpers/sd_helpers.py b/iz_helpers/sd_helpers.py index 396eb1d..3a4b543 100644 --- a/iz_helpers/sd_helpers.py +++ b/iz_helpers/sd_helpers.py @@ -70,7 +70,12 @@ def renderImg2Img( mask=mask_image, ) # p.latent_mask = Image.new("RGB", (p.width, p.height), "white") - + processed = process_images(p) + # For those that use Image grids this will make sure that ffmpeg does not crash out + if (processed.images[0].size[0] != processed.images[1].size[0]): + processed.images.pop(0) + print("\nGrid image detected applying patch") + newseed = p.seed return processed, newseed diff --git a/iz_helpers/video.py b/iz_helpers/video.py index c3d61a3..cb1c5d9 100644 --- a/iz_helpers/video.py +++ b/iz_helpers/video.py @@ -13,9 +13,8 @@ def write_video(file_path, frames, fps, reversed=True, start_frame_dupe_amount=1 if reversed == True: frames = frames[::-1] - # Get dimensions of the first frames, all subsequent has to be same sized - for k in frames: - assert (k.size == frames[0].size,"Different frame sizes found!") + # Drop missformed frames + frames = [frame for frame in frames if frame.size == frames[0].size] # Create an imageio video writer, avoid block size of 512. writer = imageio.get_writer(file_path, fps=fps, macro_block_size=None)