adding repo support

.github/workflows/instant_benchmark.yml

@@ -39,6 +39,11 @@ on:
           - none
           - table
           - cloudwatch
+      repo:
+        description: '[Do not change] The repo for runner registration'
+        required: false
+        type: string
+        default: 'djl-serving'
   workflow_call:
     inputs:
       running_template:
@@ -59,6 +64,11 @@ on:
         required: false
         type: string
         default: 'none'
+      repo:
+        description: 'The repo for runner registration'
+        required: false
+        type: string
+        default: 'djl-serving'
 
 permissions:
   id-token: write
@@ -73,10 +83,10 @@ jobs:
         run: |
           cd /home/ubuntu/djl_benchmark_script/scripts
           token=$( curl -X POST -H "Authorization: token ${{ secrets.ACTION_RUNNER_PERSONAL_TOKEN }}" \
-          https://api.github.com/repos/deepjavalibrary/djl-serving/actions/runners/registration-token \
+          https://api.github.com/repos/deepjavalibrary/${{ inputs.repo }}/actions/runners/registration-token \
           --fail \
           | jq '.token' | tr -d '"' )
-          ./start_instance.sh action_ib_${{ inputs.instance }} $token djl-serving
+          ./start_instance.sh action_ib_${{ inputs.instance }} $token ${{ inputs.repo }}
     outputs:
       gpu_instance_id: ${{ steps.create_instance.outputs.action_ib_instance_id }}