fastvideo.v1.pipelines.stages.stepvideo_encoding
#
Module Contents#
Classes#
Stage for encoding prompts using the remote caption API. |
Data#
API#
- class fastvideo.v1.pipelines.stages.stepvideo_encoding.StepvideoPromptEncodingStage(stepllm, clip)[source]#
Bases:
fastvideo.v1.pipelines.stages.base.PipelineStage
Stage for encoding prompts using the remote caption API.
This stage applies the magic string transformations and calls the remote caption service asynchronously to get:
primary prompt embeddings,
an attention mask,
and a clip embedding.
Initialization
- forward(batch: fastvideo.v1.pipelines.pipeline_batch_info.ForwardBatch, fastvideo_args) fastvideo.v1.pipelines.pipeline_batch_info.ForwardBatch [source]#