fix name builder

deepjavalibrary · Jan 8, 2024 · da79fc6 · da79fc6
1 parent 8cafed6
commit da79fc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/llm_integration_p4d.yml b/.github/workflows/llm_integration_p4d.yml
@@ -211,7 +211,7 @@ jobs:
       - name: Install pip dependencies
         run: pip3 install requests numpy
       - name: Build container name
-        run: ./serving/docker/scripts/docker_name_builder.sh tensorrt-llm ${{ github.event.inputs.djl-version }}
+        run: ./serving/docker/scripts/docker_name_builder.sh deepspeed ${{ github.event.inputs.djl-version }}
       - name: Download models and dockers
         working-directory: tests/integration
         run: |