add more info to the error message for AUTOMATIC1111#15567

IllusiveMan196 · Apr 21, 2024 · 6f4f6bf · 6f4f6bf
1 parent 367b823
commit 6f4f6bf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/modules/textual_inversion/image_embedding.py b/modules/textual_inversion/image_embedding.py
@@ -1,5 +1,6 @@
 import base64
 import json
+import os.path
 import warnings
 import logging
 
@@ -117,7 +118,7 @@ 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:
-        logger.debug('No Image data blocks found.')
+        logger.debug(f'{os.path.basename(getattr(image, "filename", "unknown image file"))}: no embedded information found.')
         return None
 
     data_block_lower = outarr[:, :black_cols[0].min(), :].astype(np.uint8)