diff --git a/.github/workflows/generate-ttnn-md.yml b/.github/workflows/generate-ttnn-md.yml
index 15a7810..fb056b3 100644
--- a/.github/workflows/generate-ttnn-md.yml
+++ b/.github/workflows/generate-ttnn-md.yml
@@ -13,19 +13,13 @@ jobs:
   generate_md:
     timeout-minutes: 120
     runs-on: ubuntu-latest
+    container:
+      image: ghcr.io/tenstorrent/tt-torch/tt-torch-ci-ubuntu-22-04:latest
+      options: --user root
+
     steps:
       - uses: actions/checkout@v4
 
-      - name: Setpyenv
-        uses: actions/setup-python@v2
-        with:
-          python-version: 3.11
-
-      - name: Install dependencies
-        run: |
-          python -m pip install --upgrade pip
-          python -m pip install -r requirements.txt
-
       - name: Set reusable strings
         id: strings
         shell: bash
diff --git a/.github/workflows/run-model-tests.yml b/.github/workflows/run-model-tests.yml
index ecb1740..22cb3b6 100644
--- a/.github/workflows/run-model-tests.yml
+++ b/.github/workflows/run-model-tests.yml
@@ -14,8 +14,8 @@ jobs:
       fail-fast: false
       matrix:
         build: [
-          {runs-on: n150, name: "run1", test_names: "stable_diffusion, Qwen, MobileNetV2, clip, flan_t5, mlpmixer, resnet, vilt, albert, codegen, glpn_kitti, mnist, resnet50, RMBG, unet_carvana, mgp-str-base, musicgen_small, segformer, torchvision, yolos"},
-          {runs-on: n150, name: "run2", test_names: "t5, whisper, autoencoder_conv, deit, gpt2, mobilenet_ssd, roberta, timm, xglm, autoencoder_linear, detr, beit, distilbert, hand_landmark, openpose, segment_anything, unet, yolov3, bert, dpr, hardnet, opt, speecht5_tts, unet_brain, yolov5, bloom, falcon, llama, perceiver_io, squeeze_bert, gpt_neo"},
+          {runs-on: n150, name: "run1", test_names: "stable_diffusion, mgp-str-base"},
+          {runs-on: n150, name: "run2", test_names: "t5"},
         ]
     runs-on:
       - ${{ matrix.build.runs-on }}