From cabc9207be2f961610d648d173bfe597d19f7c5b Mon Sep 17 00:00:00 2001 From: Yaowei Zheng Date: Tue, 17 Jun 2025 01:37:47 +0800 Subject: [PATCH] [ci] add docker version (#8390) --- .github/workflows/docker.yml | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/.github/workflows/docker.yml b/.github/workflows/docker.yml index c3c38005..90e9ef87 100644 --- a/.github/workflows/docker.yml +++ b/.github/workflows/docker.yml @@ -43,6 +43,16 @@ jobs: - name: Checkout uses: actions/checkout@v4 + - name: Set up Python + uses: actions/setup-python@v5 + with: + python-version: "3.9" + + - name: Get llamafactory version + id: version + run: | + echo "tag=$(python setup.py --version | sed 's/\.dev0//')" >> "$GITHUB_OUTPUT" + - name: Set up Docker Buildx uses: docker/setup-buildx-action@v3 @@ -61,6 +71,8 @@ jobs: build-args: | EXTRAS=metrics,deepspeed,liger-kernel push: ${{ github.event_name != 'pull_request' }} - tags: docker.io/hiyouga/llamafactory:latest + tags: | + docker.io/hiyouga/llamafactory:latest + docker.io/hiyouga/llamafactory:${{ steps.version.outputs.tag }} cache-from: type=gha cache-to: type=gha,mode=max