diff --git a/.github/workflows/perf-device-models-impl.yaml b/.github/workflows/perf-device-models-impl.yaml
index 43610aa2cfd..ff074f88f7d 100644
--- a/.github/workflows/perf-device-models-impl.yaml
+++ b/.github/workflows/perf-device-models-impl.yaml
@@ -21,14 +21,25 @@ jobs:
         ]
     name: "${{ matrix.test-info.name }} device perf"
     env:
-      TT_METAL_ENV: ${{ vars.TT_METAL_ENV }}
       ARCH_NAME: ${{ matrix.test-info.arch }}
       LOGURU_LEVEL: INFO
       LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
     runs-on: ${{ matrix.test-info.runs-on }}
+    container:
+      image: ghcr.io/tenstorrent/tt-metal/tt-metalium/ubuntu-22.04-amd64:latest
+      env:
+        TT_METAL_HOME: ${{ github.workspace }}
+        ARCH_NAME: ${{ matrix.test-info.arch }}
+        LD_LIBRARY_PATH: ${{ github.workspace }}/build/lib
+        PYTHON_ENV_DIR: ${{ github.workspace }}/python_env
+      volumes:
+        - /dev/hugepages-1G:/dev/hugepages-1G
+        - /mnt/MLPerf:/mnt/MLPerf
+      options: --device /dev/tenstorrent
     steps:
+      - name: Set safe directory for Git
+        run: git config --global --add safe.directory $GITHUB_WORKSPACE
       - uses: tenstorrent/tt-metal/.github/actions/checkout-with-submodule-lfs@main
-      - uses: ./.github/actions/ensure-active-weka-mount
       - name: Set up dynamic env vars for build
         run: |
           echo "TT_METAL_HOME=$(pwd)" >> $GITHUB_ENV