We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a37d570 commit bde00ceCopy full SHA for bde00ce
1 file changed
fastvideo/worker/multiproc_executor.py
@@ -74,7 +74,8 @@ class StreamingResult:
74
75
76
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):
+ if (fastvideo_args is not None and getattr(fastvideo_args, "output_type", None) == "latent"
78
+ and isinstance(result, torch.Tensor) and result.is_cuda):
79
return result.detach().cpu()
80
return result
81
0 commit comments