diff --git a/.github/workflows/backend.yml b/.github/workflows/backend.yml index 048e9a47b..85b8737d6 100644 --- a/.github/workflows/backend.yml +++ b/.github/workflows/backend.yml @@ -882,7 +882,7 @@ jobs: backend: "rfdetr" dockerfile: "./backend/Dockerfile.python" context: "./backend" - - build-type: 'cublas' + - build-type: 'l4t' cuda-major-version: "12" cuda-minor-version: "0" platforms: 'linux/arm64' @@ -955,13 +955,13 @@ jobs: backend: "exllama2" dockerfile: "./backend/Dockerfile.python" context: "./backend" - - build-type: 'cublas' + - build-type: 'l4t' cuda-major-version: "12" cuda-minor-version: "0" platforms: 'linux/arm64' skip-drivers: 'true' tag-latest: 'auto' - tag-suffix: '-nvidia-l4t-arm64-chatterbox' + tag-suffix: '-gpu-nvidia-l4t-arm64-chatterbox' base-image: "nvcr.io/nvidia/l4t-jetpack:r36.4.0" runs-on: 'ubuntu-24.04-arm' backend: "chatterbox"