FilesExpand file tree

master

/

node_helpers.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (31 loc) · 960 Bytes

master

/

node_helpers.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (31 loc) · 960 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

import hashlib

from comfy.cli_args import args

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

def hasher():

hashfuncs = {

"md5": hashlib.md5,

"sha1": hashlib.sha1,

"sha256": hashlib.sha256,

"sha512": hashlib.sha512

}

return hashfuncs[args.default_hashing_function]