mirror of
https://github.com/huggingface/diffusers.git
synced 2026-10-27 21:33:02 +03:00
update
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user