fastvideo.v1.pipelines.pipeline_batch_info
#
Data structures for functional pipeline processing.
This module defines the dataclasses used to pass state between pipeline components in a functional manner, reducing the need for explicit parameter passing.
Module Contents#
Classes#
Complete state passed through the pipeline execution. |
API#
- class fastvideo.v1.pipelines.pipeline_batch_info.ForwardBatch[source]#
Complete state passed through the pipeline execution.
This dataclass contains all information needed during the diffusion pipeline execution, allowing methods to update specific components without needing to manage numerous individual parameters.
- clip_embedding_neg: Optional[List[torch.Tensor]][source]#
None
- clip_embedding_pos: Optional[List[torch.Tensor]][source]#
None
- generator: Optional[Union[torch.Generator, List[torch.Generator]]][source]#
None
- image_embeds: List[torch.Tensor][source]#
‘field(…)’
- image_latent: Optional[torch.Tensor][source]#
None
- latents: Optional[torch.Tensor][source]#
None
- negative_attention_mask: Optional[List[torch.Tensor]][source]#
None
- negative_prompt_embeds: Optional[List[torch.Tensor]][source]#
None
- noise_pred: Optional[torch.Tensor][source]#
None
- prompt_attention_mask: Optional[List[torch.Tensor]][source]#
None
- prompt_embeds: List[torch.Tensor][source]#
‘field(…)’
- teacache_params: Optional[fastvideo.v1.configs.sample.teacache.TeaCacheParams | fastvideo.v1.configs.sample.teacache.WanTeaCacheParams][source]#
None
- timesteps: Optional[torch.Tensor][source]#
None