r""" Forward pass through the diffusion model Args: x (List[Tensor]): List of input video tensors, each w

r""" Args: x(Tensor): Shape [B, L, C]

1D RoPE. Args: query (torch.tensor): [B, head, seq, head_dim] pos_indices (torch.tensor): [seq,] Returns:

Matches the color of a source video chunk to a reference image and blends with the original. Args: source_chunk (torch.Tensor): The

x: A list of videos each with shape [C, T, H, W]. t: [B]. context: A list of text embeddings each with s

x: A list of videos each with shape [C, T, H, W]. t: [B]. context: A list of text embeddings each with s

XLMRobertaLarge adapted from Huggingface.