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

...

Browse Source
This commit is contained in:
Yousef Rafat 2026-04-30 11:43:53 +03:00
parent 87994368cc
commit 87514354a5
2 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
comfy_extras/nodes_convert_color_space.py
View File

@ -71,7 +71,7 @@ class ConvertColorSpace(IO.ComfyNode):
elif source_color_space == "HDR Display (PQ/Rec.2020)":
# assuming Linear Rec.2020 input. Convert to Linear Rec.709
matrix = M_2020_to_709.to(device)
matrix = M_2020_to_709.to(device=device, dtype=rgb.dtype)
rgb = pq_to_linear(rgb)
rgb = torch.matmul(rgb, matrix.T)
@ -87,7 +87,7 @@ class ConvertColorSpace(IO.ComfyNode):
elif target_color_space == "HDR Display (PQ/Rec.2020)":
# convert Gamut from Linear Rec.709 to Linear Rec.2020
rgb = torch.matmul(rgb, M_709_to_2020.to(device).T).clamp(min=0)
rgb = torch.matmul(rgb, M_709_to_2020.to(device=device, dtype=rgb.dtype).T).clamp(min=0)
rgb = linear_to_pq(rgb)
img_tensor = torch.cat([rgb, alpha], dim=-1) if has_alpha else rgb

5
comfy_extras/nodes_images.py
View File

@ -1068,10 +1068,7 @@ class SaveImageAdvanced(IO.ComfyNode):
stream.pix_fmt = av_fmt
elif file_format == "avif":
try:
stream = container.add_stream('libsvtav1', rate=1)
except Exception:
stream = container.add_stream('av1', rate=1)
stream = container.add_stream('libsvtav1', rate=1)
stream.time_base = Fraction(1, 1)