Summary: 2 instances, 2 unique

Text	Count
# TODO: update this when transformer gets converted to dataclass configs	1
# TODO replace `nonzero(as_tuple=False)` after TorchScript supports it	1