nodes/SEkINVR-ComfyUI-SaveAs
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update save_multi_format.py

Browse Source
This commit is contained in:
SEKiN.VR
2024-06-28 19:30:26 -07:00
committed by GitHub
parent ce032e5261
commit b4b376bb00
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
save_multi_format.py
+5 -2
View File
@@ -15,7 +15,7 @@ class SaveMultiFormatImage:
return {
"required": {
"images": ("IMAGE",),
"filename_prefix": ("STRING", {"default": "image"}),
"filename_prefix": ("STRING", {"default": "icon"}),
"format": (["png", "jpg", "webp", "ico"],),
"quality": ("INT", {"default": 95, "min": 1, "max": 100, "step": 1}),
"ico_size_preset": (list(s.ICO_SIZES.keys()),),
@@ -42,7 +42,10 @@ class SaveMultiFormatImage:
for i, image in enumerate(images):
img = Image.fromarray((255. * image.cpu().numpy()).astype(np.uint8))
filename = self._generate_filename(save_dir, filename_prefix, i, format)
if format == "ico":
filename = self._generate_filename(save_dir, "icon", i, format)
else:
filename = self._generate_filename(save_dir, filename_prefix, i, format)
if format == "ico":
self._save_ico(img, filename, ico_sizes)