Model Demos

This page demonstrates how to run the tt-metalium model demos using the tt-installer tool. The tt-installer can be used to download a container for running the tt-metalium demos. This container possesses a full build of the tt-metalium project, including the demo source code.

⚠️ NOTE: this page assumes you have already used tt-installer to install the system dependencies as shown in the starting guide

1. Downloading and installing the tt-metalium models container

Execute the following command to download and install the tt-metalium models container:

/bin/bash -c "$(curl -fsSL https://github.com/tenstorrent/tt-installer/releases/latest/download/install.sh)" --no-install-kmd --no-install-hugepages --no-install-metalium-container --install-metalium-models-container --no-install-tt-flash --no-install-tt-topology --update-firmware="off" --reboot-option="never" --mode-non-interactive

2. Starting the container and running a simple program

To use the models container, execute this command to create an interactive shell with all configuration taken care of:

tt-metalium-models

NOTE: this container is ephemeral so all changes made inside will be lost when the container is stopped

Run the following simple program to perform a simple exponentiation + matrix multiplication operation:

# this should be run inside the container
python ttnn/ttnn/examples/usage/run_op_on_device.py

3. Run more model demos

The full list of model demos to run can be found in the tt-metal GitHub repository. Scroll down that page to find the table of model demos for the following model types:

  • LLMs

  • Speech-To-Text

  • Diffusion

  • Image Classification

  • Vision Transformers

  • Object Detection

  • Segmentation

  • NLPs

Clicking on each model’s name will navigate to the demo page which contains all instructions for running that demo.

4. Exiting the container

To exit the container, simply run this command:

exit