Graphing Torch DiT_XL_2 With TTNN

Note

This TT-NN tutorial is available on Grayskull only. We are working on compatibility on other architectures.

• Build a graph of a pytorch based model
  • Clone the library from https://github.com/facebookresearch/DiT.git
  • Download DiT-XL/2 Models
  • Sample from Pre-trained DiT Models and build the graph
  • Display the graph