diff --git a/.github/workflows/main.yml b/.github/workflows/main.yml index 9446e03a1..1e570fce0 100644 --- a/.github/workflows/main.yml +++ b/.github/workflows/main.yml @@ -4,7 +4,6 @@ on: push: branches: - "main" - - "dev" jobs: build-axolotl: @@ -72,6 +71,7 @@ jobs: python_version: "3.10" pytorch: 2.0.1 axolotl_extras: + is_latest: true - cuda: 118 cuda_version: 11.8.0 python_version: "3.9" @@ -102,5 +102,7 @@ jobs: CUDA=${{ matrix.cuda }} file: ./docker/Dockerfile-runpod push: ${{ github.event_name != 'pull_request' }} - tags: ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }} + tags: | + ${{ steps.metadata.outputs.tags }}-py${{ matrix.python_version }}-cu${{ matrix.cuda }}-${{ matrix.pytorch }}${{ matrix.axolotl_extras != '' && '-' || '' }}${{ matrix.axolotl_extras }} + ${{ (matrix.is_latest) && format('{0}:latest', steps.metadata.outputs.image) || '' }} labels: ${{ steps.metadata.outputs.labels }}