wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-18 13:07:28 +08:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

interept_as as combo

Browse Source
This commit is contained in:
Yousef Rafat 2026-04-29 14:06:40 +03:00
parent 14e114a936
commit cde0936e42
1 changed files with 13 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
comfy_extras/nodes_images.py
View File

@ -963,11 +963,17 @@ class SaveImageAdvanced(IO.ComfyNode):
default="ComfyUI",
tooltip="The prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.",
),
IO.Combo.Input(
"interept_as",
options=["Raw/Data", "sRGB"],
default="sRGB",
advanced=True,
),
IO.DynamicCombo.Input(
"format",
options=[
IO.DynamicCombo.Option(
"png",
"png",
[
IO.Combo.Input(
"bit_depth",
@ -975,12 +981,6 @@ class SaveImageAdvanced(IO.ComfyNode):
default="8-bit",
advanced=True,
),
IO.Combo.Input(
"color_space",
options=["Raw/Data", "sRGB"],
default="sRGB",
advanced=True,
),
],
),
IO.DynamicCombo.Option(
@ -992,12 +992,6 @@ class SaveImageAdvanced(IO.ComfyNode):
default="8-bit",
advanced=True,
),
IO.Combo.Input(
"color_space",
options=["sRGB"],
default="sRGB",
advanced=True,
),
],
),
IO.DynamicCombo.Option(
@ -1005,14 +999,8 @@ class SaveImageAdvanced(IO.ComfyNode):
[
IO.Combo.Input(
"bit_depth",
options=["16-bit (half-float)", "32-bit"],
default="16-bit (half-float)",
advanced=True,
),
IO.Combo.Input(
"color_space",
options=["Linear", "Raw/Data"],
default="Linear",
options=["16-bit", "32-bit"],
default="16-bit",
advanced=True,
),
],
@ -1027,22 +1015,15 @@ class SaveImageAdvanced(IO.ComfyNode):
)
@classmethod
def execute(
cls,
images: Input.Image,
filename_prefix: str,
format: dict,
embed_workflow: bool,
prompt=None,
extra_pnginfo=None
) -> IO.NodeOutput:
def execute(cls, images, filename_prefix: str, format: dict, embed_workflow: bool, prompt=None, extra_pnginfo=None) -> IO.NodeOutput:
output_dir = folder_paths.get_output_directory()
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.get_save_image_path(filename_prefix, output_dir, images[0].shape[1], images[0].shape[0])
full_output_folder, filename, counter, subfolder, filename_prefix = folder_paths.\
get_save_image_path(filename_prefix, output_dir, images[0].shape[1], images[0].shape[0])
results = list()
for batch_number, image in enumerate(images):
# get widget values from dynamic combo
extension = format["format"]
file_format = format["format"]
bit_depth = format["bit_depth"]
color_space = format["color_space"]
@ -1059,12 +1040,6 @@ class SaveImageAdvanced(IO.ComfyNode):
if file_format in ["png", "exr", "avif"]:
# safe bit downcasting
if (file_format == "png" or file_format == "avif") and bit_depth == "32-bit":
bit_depth = "16-bit"
if file_format == "exr" and bit_depth == "8-bit":
bit_depth = "16-bit"
if bit_depth == "32-bit":
img_np = img_tensor.cpu().numpy().astype(np.float32)
av_fmt = 'gbrapf32le' if has_alpha else 'gbrpf32le'