minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] feat: speed up lossless realtime rgb transport (#27236)

Browse Source
This commit is contained in:
Mick
2026-06-04 16:59:23 +08:00
committed by GitHub
parent ef170c27b6
commit 9e2ad6054c
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/utils/realtime_video.py
+5 -1
View File
@@ -27,6 +27,7 @@ WEBP_FRAME_CONTENT_TYPE = "image/webp"
JPEG_FRAME_CONTENT_TYPE = "image/jpeg" JPEG_FRAME_CONTENT_TYPE = "image/jpeg"
RAW_RGB_CHANNELS = 3 RAW_RGB_CHANNELS = 3
RAW_RGBA_CHANNELS = 4 RAW_RGBA_CHANNELS = 4
_RAW_RGB_DELTA_GZIP_LEVEL = 0
def build_delta_gzip_raw_rgb_payload( def build_delta_gzip_raw_rgb_payload(
@@ -46,7 +47,10 @@ def build_delta_gzip_raw_rgb_payload(
if reference_frame is not None if reference_frame is not None
else None else None
) )
compressor = zlib.compressobj(level=1, method=zlib.DEFLATED, wbits=31) # keep gzip framing for lossless transport without spending realtime budget on compression
compressor = zlib.compressobj(
level=_RAW_RGB_DELTA_GZIP_LEVEL, method=zlib.DEFLATED, wbits=31
)
compressed_chunks = [] compressed_chunks = []
for frame in frames: for frame in frames:
if len(frame) != frame_size: if len(frame) != frame_size: