diff --git a/tensorrt_llm/media/encoding.py b/tensorrt_llm/media/encoding.py index 0e2e32490909..5a5484f79e1f 100644 --- a/tensorrt_llm/media/encoding.py +++ b/tensorrt_llm/media/encoding.py @@ -181,7 +181,7 @@ def encode_video( stdout=subprocess.PIPE, stderr=subprocess.PIPE, ) - raw_frames = video_np.tobytes() + raw_frames = memoryview(video_np.reshape(-1)) _, stderr = process.communicate(input=raw_frames) if process.returncode != 0: raise RuntimeError(f"ffmpeg encoding failed: {stderr.decode()}")