#0: Fix demo pipeline workflow

tenstorrent · Dec 11, 2024 · d227192 · d227192
1 parent 289defa
commit d227192
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.github/workflows/tg-demo-tests-impl.yaml b/.github/workflows/tg-demo-tests-impl.yaml
@@ -12,7 +12,7 @@ jobs:
           { name: "TG Llama3 demo tests",arch: wormhole_b0, model: 'llama3', timeout: 90, owner_id: U06F3ER8X9A}, # Stuti Raizada
           { name: "TG Falcon7b demo tests", arch: wormhole_b0, model: falcon7b, timeout: 30, owner_id: U05RWH3QUPM}, #Salar Hosseini
         ]
-    args: '--tt-arch wormhole_b0 --pipeline-type unit_tg_device --dispatch-mode ""'
+        args: '--tt-arch wormhole_b0 --pipeline-type unit_tg_device --dispatch-mode "" --model ${{ matrix.test-group.model }}'
     name: ${{ matrix.test-group.name }}
     env:
       TT_METAL_ENV: ${{ vars.TT_METAL_ENV }}
@@ -43,4 +43,4 @@ jobs:
           cd $TT_METAL_HOME
           export PYTHONPATH=$TT_METAL_HOME
           source ${{ github.workspace }}/tests/scripts/run_tests.sh
-          ${{ args }} --model ${{ matrix.test-group.model }}
+          ${{ matrix.args }}