fastvideo.v1.worker.gpu_worker

fastvideo.v1.worker.gpu_worker#

Module Contents#

Classes#

Worker

Functions#

run_worker_process

Data#

CYAN

RESET

logger

API#

fastvideo.v1.worker.gpu_worker.CYAN[source]#

‘\x1b[1;36m’

fastvideo.v1.worker.gpu_worker.RESET[source]#

‘\x1b[0;0m’

class fastvideo.v1.worker.gpu_worker.Worker(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs, local_rank: int, rank: int, pipe: multiprocessing.connection.Connection, master_port: int)[source]#

Initialization

event_loop() → None[source]#

Event loop for the worker.

execute_forward(forward_batch: fastvideo.v1.pipelines.ForwardBatch, fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs) → fastvideo.v1.pipelines.ForwardBatch[source]#
init_device() → None[source]#

Initialize the device for the worker.

set_lora_adapter(lora_nickname: str, lora_path: str) → None[source]#
shutdown() → dict[str, Any][source]#

Gracefully shut down the worker process

fastvideo.v1.worker.gpu_worker.logger[source]#

‘init_logger(…)’

fastvideo.v1.worker.gpu_worker.run_worker_process(fastvideo_args: fastvideo.v1.fastvideo_args.FastVideoArgs, local_rank: int, rank: int, pipe: multiprocessing.connection.Connection, master_port: int)[source]#
Contents