diff --git a/.github/workflows/test-code-build.yml b/.github/workflows/test-code-build.yml
index 3c2a359..5d03f30 100644
--- a/.github/workflows/test-code-build.yml
+++ b/.github/workflows/test-code-build.yml
@@ -5,6 +5,7 @@ on:
 jobs:
   build_wheels:
     name: Build wheels and Upload
+    timeout-minutes: 3600
     # runs-on: windows-latest
     runs-on: codebuild-flash-attention-pre-build-wheel-windows-${{ github.run_id }}-${{ github.run_attempt }}
     steps:
@@ -26,6 +27,7 @@ jobs:
 
       - name: Install VisualStudio BuildTools
         shell: pwsh
+        timeout-minutes: 1200
         run: |
           choco install -y visualstudio2022buildtools `
             --version=117.14.1 --params "--add Microsoft.VisualStudio.Component.VC.Tools.x86.x64 --includeRecommended --includeOptional"
@@ -46,6 +48,7 @@ jobs:
           pip install -U pip setuptools==75.8.0 wheel setuptools packaging psutil ninja numpy
 
       - name: Build wheels
+        timeout-minutes: 1200
         shell: pwsh
         env:
           MAX_JOBS: 4