Spaces:
Sleeping
Sleeping
from PIL import ImageFile, UnidentifiedImageError | |
def conditioning_set_values(conditioning, values={}): | |
c = [] | |
for t in conditioning: | |
n = [t[0], t[1].copy()] | |
for k in values: | |
n[1][k] = values[k] | |
c.append(n) | |
return c | |
def pillow(fn, arg): | |
prev_value = None | |
try: | |
x = fn(arg) | |
except (OSError, UnidentifiedImageError, ValueError): #PIL issues #4472 and #2445, also fixes ComfyUI issue #3416 | |
prev_value = ImageFile.LOAD_TRUNCATED_IMAGES | |
ImageFile.LOAD_TRUNCATED_IMAGES = True | |
x = fn(arg) | |
finally: | |
if prev_value is not None: | |
ImageFile.LOAD_TRUNCATED_IMAGES = prev_value | |
return x | |