From 605988f520362886b965d65940a4e32a68f2f68c Mon Sep 17 00:00:00 2001 From: Aptronymist <108482020+Aptronymist@users.noreply.github.com> Date: Fri, 11 Aug 2023 19:51:42 -0400 Subject: [PATCH] Removed annoyance Took out the print statement 'no image data blocks found' as it happens frequently as embeddings rarely have image data. --- modules/textual_inversion/image_embedding.py | 1 - 1 file changed, 1 deletion(-) diff --git a/modules/textual_inversion/image_embedding.py b/modules/textual_inversion/image_embedding.py index aca25bc9a..7c9fe881d 100644 --- a/modules/textual_inversion/image_embedding.py +++ b/modules/textual_inversion/image_embedding.py @@ -113,7 +113,6 @@ def extract_image_data_embed(image): outarr = crop_black(np.array(image.convert('RGB').getdata()).reshape(image.size[1], image.size[0], d).astype(np.uint8)) & 0x0F black_cols = np.where(np.sum(outarr, axis=(0, 2)) == 0) if black_cols[0].shape[0] < 2: - print('No Image data blocks found.') return None data_block_lower = outarr[:, :black_cols[0].min(), :].astype(np.uint8)