1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-10-27 21:33:02 +03:00
This commit is contained in:
Dhruv Nair
2024-01-18 08:29:19 +00:00
parent e601a992b6
commit bd1ac23d1d

View File

@@ -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