From bd1ac23d1d6badbbfbf91c6efc12a2f2daf662a9 Mon Sep 17 00:00:00 2001 From: Dhruv Nair Date: Thu, 18 Jan 2024 08:29:19 +0000 Subject: [PATCH] update --- .../text_to_video_synthesis/pipeline_text_to_video_synth.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py b/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py index d7c1322aed..b665d4c89c 100644 --- a/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py +++ b/src/diffusers/pipelines/text_to_video_synthesis/pipeline_text_to_video_synth.py @@ -73,7 +73,7 @@ def tensor2vid(video: torch.Tensor, processor, output_type="np"): outputs = np.stack(outputs) if output_type == "pt": - outputs = torch.cat(outputs, dim=0) + outputs = torch.stack(outputs) return outputs