fastvideo.worker.multiproc_executor

fastvideo.worker.multiproc_executor#

Module Contents#

Classes#

MultiprocExecutor

Data#

logger

API#

class fastvideo.worker.multiproc_executor.MultiprocExecutor(fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs)[source]#

Bases: fastvideo.worker.executor.Executor

collective_rpc(method: str | collections.abc.Callable, timeout: float | None = None, args: tuple = (), kwargs: dict | None = None) → list[Any][source]#
execute_forward(forward_batch: fastvideo.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args: fastvideo.fastvideo_args.FastVideoArgs) → fastvideo.pipelines.pipeline_batch_info.ForwardBatch[source]#
merge_lora_weights() → None[source]#
set_lora_adapter(lora_nickname: str, lora_path: str | None = None) → None[source]#
shutdown() → None[source]#

Properly shut down the executor and its workers

unmerge_lora_weights() → None[source]#
fastvideo.worker.multiproc_executor.logger[source]#

‘init_logger(…)’

Contents