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

bytesio -> tempfile

Browse Source
This commit is contained in:
Yousef Rafat 2026-04-24 23:27:59 +03:00
parent 8fff8814ad
commit ebb9acf3cf
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
comfy_extras/nodes_save_advanced.py
View File

@ -1,4 +1,3 @@
import io
import av import av
import os import os
import json import json
@ -7,6 +6,7 @@ import struct
import zlib import zlib
import logging import logging
import numpy as np import numpy as np
import tempfile
import folder_paths import folder_paths
from comfy_api.latest import IO from comfy_api.latest import IO
from typing_extensions import override from typing_extensions import override
@ -181,9 +181,10 @@ class SaveImageAdvanced(IO.ComfyNode):
img_np = (img_tensor * 255.0).clamp(0, 255).to(torch.int32).cpu().numpy().astype(np.uint8) img_np = (img_tensor * 255.0).clamp(0, 255).to(torch.int32).cpu().numpy().astype(np.uint8)
av_fmt = 'rgba' if has_alpha else 'rgb24' av_fmt = 'rgba' if has_alpha else 'rgb24'
memory_buffer = io.BytesIO() fd, tmp_path = tempfile.mkstemp(suffix=f".{file_format}")
os.close(fd)
container_format = "image2" if file_format in ["png", "exr"] else "avif" container_format = "image2" if file_format in ["png", "exr"] else "avif"
container = av.open(memory_buffer, mode='w', format=container_format) container = av.open(tmp_path, mode='w', format=container_format)
if file_format == "exr": if file_format == "exr":
stream = container.add_stream('exr', rate=1) stream = container.add_stream('exr', rate=1)
@ -223,7 +224,9 @@ class SaveImageAdvanced(IO.ComfyNode):
container.close() container.close()
final_bytes = memory_buffer.getvalue() with open(tmp_path, "rb") as f:
final_bytes = f.read()
os.remove(tmp_path)
if embed_workflow and not args.disable_metadata: if embed_workflow and not args.disable_metadata:
if file_format == "png": if file_format == "png":