Mirror
/
stable-diffusion-webui
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15567 from AUTOMATIC1111/no-image-data-blocks-debug 

Hide 'No Image data blocks found.' message
This commit is contained in:
AUTOMATIC1111 2024-04-21 07:09:27 +03:00 committed by GitHub
parent c8ac42aad1 d212fb59fe
commit 367b823466
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
modules/textual_inversion/image_embedding.py
View File

@ -1,12 +1,15 @@
import base64 import base64
import json import json
import warnings import warnings
import logging
import numpy as np import numpy as np
import zlib import zlib
from PIL import Image, ImageDraw from PIL import Image, ImageDraw
import torch import torch
logger = logging.getLogger(__name__)
class EmbeddingEncoder(json.JSONEncoder): class EmbeddingEncoder(json.JSONEncoder):
def default(self, obj): def default(self, obj):
@ -114,7 +117,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 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) black_cols = np.where(np.sum(outarr, axis=(0, 2)) == 0)
if black_cols[0].shape[0] < 2: if black_cols[0].shape[0] < 2:
print('No Image data blocks found.') logger.debug('No Image data blocks found.')
return None return None
data_block_lower = outarr[:, :black_cols[0].min(), :].astype(np.uint8) data_block_lower = outarr[:, :black_cols[0].min(), :].astype(np.uint8)