minor fix to the instance spinning (#2606)

deepjavalibrary · May 16, 2023 · 9e2bc98 · 9e2bc98
1 parent 1e4598e
commit 9e2bc98
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.github/workflows/build_triton_and_ft.yml b/.github/workflows/build_triton_and_ft.yml
@@ -26,6 +26,7 @@ jobs:
         with:
           python-version: '3.x'
       - name: Build Triton Binary
+        shell: 'script -q -e -c "bash --noprofile --norc -eo pipefail {0}"'
         run: |
           python3 build.py --enable-logging --enable-metrics --enable-stats --enable-cpu-metrics
       - name: Configure AWS Credentials
@@ -48,10 +49,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/djl/actions/runners/registration-token \
           --fail \
           | jq '.token' | tr -d '"' )
-          ./start_instance.sh action_cpu $token djl-serving
+          ./start_instance.sh action_cpu $token djl
     outputs:
       cpu_instance_id: ${{ steps.create_cpu.outputs.action_cpu_instance_id }}
 

diff --git a/tools/scripts/build_ft_deps.sh b/tools/scripts/build_ft_deps.sh
@@ -5,7 +5,7 @@ set -ex
 FT_VERSION=$1
 NVIDIA_TRITON_SERVER_VERSION=$2
 
-apt-get install -y rapidjson-dev
+apt-get update && apt-get install -y rapidjson-dev
 
 pushd /tmp