fastvideo.v1.layers.utils
#
Utility methods for model layers.
Module Contents#
Functions#
API#
- fastvideo.v1.layers.utils.get_token_bin_counts_and_mask(tokens: torch.Tensor, vocab_size: int, num_seqs: int) Tuple[torch.Tensor, torch.Tensor] [source]#