fastvideo.layers.utils

Utility methods for model layers.

Module Contents

Functions

get_token_bin_counts_and_mask

API

fastvideo.layers.utils.get_token_bin_counts_and_mask(tokens: torch.Tensor, vocab_size: int, num_seqs: int) → tuple[torch.Tensor, torch.Tensor]