Skip to content

nestedtensor.py on Colab building from master  #2196

Open
@alex-rakowski

Description

@alex-rakowski

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.

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions