wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-05-26 08:57:26 +08:00
Code Issues Actions 22 Packages Projects Releases Wiki Activity

default pyav build doesn't support float16 exr

Browse Source
This commit is contained in:
Yousef Rafat 2026-04-28 22:56:42 +03:00
parent 29653e7fcc
commit 34d7a5da73
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
comfy_extras/nodes_save_advanced.py
View File

@ -171,8 +171,9 @@ class SaveImageAdvanced(IO.ComfyNode):
av_fmt = 'gbrapf32le' if has_alpha else 'gbrpf32le' av_fmt = 'gbrapf32le' if has_alpha else 'gbrpf32le'
elif bit_depth == "16-bit": elif bit_depth == "16-bit":
if file_format == "exr": if file_format == "exr":
img_np = img_tensor.cpu().numpy().astype(np.float16) # default pyav build doesn't come with a codec for float16 exr format
av_fmt = 'gbrapf16le' if has_alpha else 'gbrpf16le' img_np = img_tensor.cpu().numpy().astype(np.float32)
av_fmt = 'gbrapf32le' if has_alpha else 'gbrpf32le'
else: else:
img_np = (img_tensor * 65535.0).clamp(0, 65535).to(torch.int32).cpu().numpy().astype(np.uint16) img_np = (img_tensor * 65535.0).clamp(0, 65535).to(torch.int32).cpu().numpy().astype(np.uint16)
av_fmt = 'rgba64le' if has_alpha else 'rgb48le' av_fmt = 'rgba64le' if has_alpha else 'rgb48le'