When running the nested tensors tutorial on google colab builds from master instead of main. The master branch version is non-functional, main branch version appears to work correctly:
- nested tensors created with
torch.nested_tensor instead of torch.nested.nested_tensor
- the
mha_netsed function handle batch size inference incorrectly.