diff --git a/.github/workflows/fast-dispatch-full-regressions-and-models-impl.yaml b/.github/workflows/fast-dispatch-full-regressions-and-models-impl.yaml index eb55fb592cc..f0e4e07966a 100644 --- a/.github/workflows/fast-dispatch-full-regressions-and-models-impl.yaml +++ b/.github/workflows/fast-dispatch-full-regressions-and-models-impl.yaml @@ -112,7 +112,7 @@ jobs: fail-fast: false matrix: card: [N150, N300] - model: [common_models, functional_unet, llama3.2-1B, mamba, mistral7b, mistral7b_eth, resnet50] + model: [common_models, functional_unet, llama3.2-1B, mamba, mistral7b, mistral7b_eth, resnet50, yolov4] name: Nightly ${{ matrix.card }} ${{ matrix.model }} env: ARCH_NAME: wormhole_b0 diff --git a/models/demos/yolov4/tests/yolov4_perfomant.py b/models/demos/yolov4/tests/yolov4_perfomant.py index 620edd78dc1..d35644da15c 100644 --- a/models/demos/yolov4/tests/yolov4_perfomant.py +++ b/models/demos/yolov4/tests/yolov4_perfomant.py @@ -1,3 +1,7 @@ +# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + import pytest import torch import ttnn diff --git a/tests/scripts/run_python_model_tests.sh b/tests/scripts/run_python_model_tests.sh index 0643861ec26..71a286e125b 100755 --- a/tests/scripts/run_python_model_tests.sh +++ b/tests/scripts/run_python_model_tests.sh @@ -35,7 +35,7 @@ run_python_model_tests_wormhole_b0() { # higher sequence lengths and different formats trigger memory issues pytest models/demos/falcon7b_common/tests/unit_tests/test_falcon_matmuls_and_bmms_with_mixed_precision.py -k "seq_len_128 and in0_BFLOAT16-in1_BFLOAT8_B-out_BFLOAT16-weights_DRAM" pytest tests/ttnn/integration_tests/resnet/test_ttnn_functional_resnet50_new.py -k "pretrained_weight_false" - pytest models/experimental/yolov4/demo/demo.py -k "pretrained_weight_false" + pytest models/demo/yolov4/demo/demo.py -k "pretrained_weight_false" # Unet Shallow WH_ARCH_YAML=wormhole_b0_80_arch_eth_dispatch.yaml pytest -svv models/experimental/functional_unet/tests/test_unet_model.py