Skip to content

Commit 8cd37ea

Browse files
Add pytorch model demos(port from tt-forge-fe) & add tt-forge-models as submodule
1 parent f749d99 commit 8cd37ea

File tree

4 files changed

+24
-0
lines changed

4 files changed

+24
-0
lines changed

.gitmodules

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,3 @@
1+
[submodule "third_party/tt_forge_models"]
2+
path = third_party/tt_forge_models
3+
url = https://github.com/tenstorrent/tt-forge-models.git
Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
# SPDX-FileCopyrightText: © 2025 Tenstorrent AI ULC
2+
# SPDX-License-Identifier: Apache-2.0
3+
4+
# Efficientnet Demo Script
5+
6+
import forge
7+
from ....third_party.tt_forge_models.efficientnet.pytorch import ModelLoader
8+
9+
# Load model and input
10+
model = ModelLoader.load_model()
11+
inputs = ModelLoader.load_inputs()
12+
13+
# Compile the model using Forge
14+
compiled_model = forge.compile(model, sample_inputs=[inputs])
15+
16+
# Run inference on Tenstorrent device
17+
output = compiled_model(inputs)
18+
19+
# Post-process the output
20+
# ModelLoader.print_cls_results(output)

third_party/__init__.py

Whitespace-only changes.

third_party/tt_forge_models

Submodule tt_forge_models added at 3b8b567

0 commit comments

Comments
 (0)