[Perf][Model] Optimize LTX video output transport (#7000)
Signed-off-by: mglyn <1203789601@qq.com>
This commit is contained in:
1 parent
8d98cc0cb7
commit
175dec11be
5 files changed
+315
-3
No files matched your search
@@ -550,6 +550,13 @@ def get_ltx2_post_process_func(od_config: Any):
|
||||
if not (isinstance(output, tuple) and len(output) == 2):
|
||||
return output
|
||||
video, audio = output
|
||||
if isinstance(video, torch.Tensor) and video.dtype == torch.uint8:
|
||||
if video.ndim != 5 or video.shape[-1] != 3 or not video.is_contiguous():
|
||||
raise ValueError(
|
||||
"LTX uint8 video output must be contiguous BTHWC RGB, "
|
||||
f"got shape={tuple(video.shape)} stride={video.stride()}"
|
||||
)
|
||||
video = video.detach().cpu().numpy()
|
||||
if isinstance(audio, torch.Tensor):
|
||||
audio = audio.detach().cpu()
|
||||
result: dict[str, Any] = {"video": video, "audio": audio}
|
||||
|
||||
Reference in new issue
Block a user