diff --git a/.github/workflows/nightly_ondemand.yml b/.github/workflows/nightly_ondemand.yml
index ff743f384..476456fa1 100644
--- a/.github/workflows/nightly_ondemand.yml
+++ b/.github/workflows/nightly_ondemand.yml
@@ -8,6 +8,11 @@ on:
     - cron: '0 16 * * 5'
   workflow_dispatch:
     inputs:
+      pytorch_repo:
+        required: false
+        type: string
+        default: 'https://github.com/pytorch/pytorch.git'
+        description: Pytorch repo
       pytorch:
         required: false
         type: string
@@ -107,8 +112,8 @@ jobs:
           pwd
           cd ../ && rm -rf pytorch
           source activate e2e_ci
-          git clone https://github.com/DDEle/pytorch
-          cd pytorch && git checkout onednn_graph_sdpa
+          git clone ${{ inputs.pytorch_repo }} pytorch
+          cd pytorch && git checkout $(echo ${{ env.pytorch }} |awk '{print $1}') 
           # apply PRs for stock pytorch
           pip install requests
           pip install botocore