avhubert/sequence_generator.py (1 line): - line 460: # TODO replace `nonzero(as_tuple=False)` after TorchScript supports it avhubert/decoder.py (1 line): - line 93: # TODO: update this when transformer gets converted to dataclass configs