xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 21:27:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: 🐛 limit loadImage dropdown to supported formats

Browse Source
This commit is contained in:
Mel Massadian 2024-07-30 21:54:49 +02:00
parent 86c5970ac0
commit fb0bd6dd31
No known key found for this signature in database
GPG Key ID: 1DFF0E9F71C7C85A
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
nodes.py
View File

@ -9,6 +9,7 @@ import math
import time import time
import random import random
import logging import logging
from pathlib import Path
from PIL import Image, ImageOps, ImageSequence, ImageFile from PIL import Image, ImageOps, ImageSequence, ImageFile
from PIL.PngImagePlugin import PngInfo from PIL.PngImagePlugin import PngInfo
@ -1523,8 +1524,9 @@ class PreviewImage(SaveImage):
class LoadImage: class LoadImage:
@classmethod @classmethod
def INPUT_TYPES(s): def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory() input_dir = Path(folder_paths.get_input_directory())
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))] pillow_formats = {ext.lower() for ext in Image.registered_extensions().keys()}
files = sorted([f.name for f in input_dir.iterdir() if f.is_file() and f.suffix.lower() in pillow_formats])
return {"required": return {"required":
{"image": (sorted(files), {"image_upload": True})}, {"image": (sorted(files), {"image_upload": True})},
} }