xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-11-12 11:46:37 +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 random
import logging
from pathlib import Path
from PIL import Image, ImageOps, ImageSequence, ImageFile
from PIL.PngImagePlugin import PngInfo
@ -1523,8 +1524,9 @@ class PreviewImage(SaveImage):
class LoadImage:
@classmethod
def INPUT_TYPES(s):
input_dir = folder_paths.get_input_directory()
files = [f for f in os.listdir(input_dir) if os.path.isfile(os.path.join(input_dir, f))]
input_dir = Path(folder_paths.get_input_directory())
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":
{"image": (sorted(files), {"image_upload": True})},
}