Skip to content

Commit bde00ce

Browse files
committed
Format latent handoff helper
1 parent a37d570 commit bde00ce

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

fastvideo/worker/multiproc_executor.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,8 @@ class StreamingResult:
7474

7575

7676
def _prepare_worker_output_for_parent(result: Any, fastvideo_args: FastVideoArgs | None) -> Any:
77-
if (fastvideo_args is not None and getattr(fastvideo_args, "output_type", None) == "latent" and isinstance(result, torch.Tensor) and result.is_cuda):
77+
if (fastvideo_args is not None and getattr(fastvideo_args, "output_type", None) == "latent"
78+
and isinstance(result, torch.Tensor) and result.is_cuda):
7879
return result.detach().cpu()
7980
return result
8081

0 commit comments

Comments
 (0)