Skip to content

Commit

Permalink
Bring up and optimize Whisper model on Wormhole/Blackhole, clean up d…
Browse files Browse the repository at this point in the history
…emo and tests, delete old obsolete versions (#18629)

Signed-off-by: Salar Hosseini <skhorasgani@tenstorrent.com>
  • Loading branch information
skhorasganiTT authored Mar 6, 2025
1 parent e4dc25d commit 07567d1
Show file tree
Hide file tree
Showing 61 changed files with 1,665 additions and 6,592 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -88,7 +88,7 @@ jobs:
fail-fast: false
matrix:
card: [N150, N300]
model: [common_models, functional_unet, llama3.2-1B, qwen, mistral7b, mistral7b_eth, resnet50, yolov4]
model: [common_models, functional_unet, llama3.2-1B, qwen, mistral7b, mistral7b_eth, resnet50, yolov4, whisper]
name: Nightly ${{ matrix.card }} ${{ matrix.model }}
env:
ARCH_NAME: wormhole_b0
Expand Down
1 change: 1 addition & 0 deletions CODEOWNERS
Original file line number Diff line number Diff line change
Expand Up @@ -177,6 +177,7 @@ models/demos/t3000/llama3_70b @cglagovichTT @uaydonat @johanna-rock-tt @djordje-
models/demos/t3000/mixtral8x7b @yieldthought @mtairum @uaydonat
models/demos/tg/llama3_70b @cglagovichTT @uaydonat @johanna-rock-tt @djordje-tt @kpaigwar
models/demos/tg/falcon7b @skhorasganiTT @djordje-tt @uaydonat
models/demos/whisper @skhorasganiTT @uaydonat
models/demos/grayskull @uaydonat
models/demos/yolov4 @dvartaniansTT @mbahnasTT @tenstorrent/metalium-developers-convolutions
models/demos/wormhole/yolov4 @dvartaniansTT @mbahnasTT @tenstorrent/metalium-developers-convolutions
Expand Down
33 changes: 33 additions & 0 deletions models/demos/whisper/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,33 @@
# Whisper Demo (Wormhole & Blackhole)

## How to Run

If running on Wormhole N300 (not required for N150 or Blackhole), the following environment variable needs to be set since Whisper requires at least 8x8 core grid size:

```sh
export WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml
```

### Conditional Generation

To run the conditional generation demo with custom inputs:
```sh
pytest --disable-warnings --input-path="models/demos/whisper/demo/dataset/conditional_generation" models/demos/whisper/demo/demo.py::test_demo_for_conditional_generation
```

To run the conditional generation demo with inputs from the `hf-internal-testing/librispeech_asr_dummy` dataset:
```sh
pytest --disable-warnings models/demos/whisper/demo/demo.py::test_demo_for_conditional_generation_dataset
```

### Audio Classification

To run the audio classification demo with custom inputs:
```sh
pytest --disable-warnings --input-path="models/demos/whisper/demo/dataset/audio_classification" models/demos/whisper/demo/demo.py::test_demo_for_audio_classification
```

To run the audio classification demo with inputs from the `google/fleurs` dataset:
```sh
pytest --disable-warnings models/demos/whisper/demo/demo.py::test_demo_for_audio_classification_dataset
```
Loading

0 comments on commit 07567d1

Please sign in to comment.