From 7e2883aa2469d8f3d4818a73a10eec6a78074184 Mon Sep 17 00:00:00 2001 From: Varshith Bathini Date: Thu, 11 Dec 2025 05:35:56 +0000 Subject: [PATCH] fix: debug log --- runner/app/live/pipelines/streamdiffusion/pipeline.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/runner/app/live/pipelines/streamdiffusion/pipeline.py b/runner/app/live/pipelines/streamdiffusion/pipeline.py index 22e6086e6..290fd49e7 100644 --- a/runner/app/live/pipelines/streamdiffusion/pipeline.py +++ b/runner/app/live/pipelines/streamdiffusion/pipeline.py @@ -97,6 +97,9 @@ def process_tensor_sync(self, img_tensor: torch.Tensor): if _is_bchw_format(out_tensor): out_tensor = out_tensor.permute(0, 2, 3, 1) + if out_tensor.dim() != 4 or out_tensor.shape[0] != 1 or out_tensor.shape[-1] != 3: + logging.error(f"[StreamDiffusion] Invalid output tensor shape for encoder: {out_tensor.shape}. Expected (1, H, W, 3). Params: {self.params}") + return out_tensor async def get_processed_video_frame(self) -> VideoOutput: