4747 - { python_version: "3.11", pytorch_version: "2.2", cuda_version: "12.1.0" }
4848 - { python_version: "3.11", pytorch_version: "2.3", cuda_version: "12.1.0" }
4949 - { python_version: "3.11", pytorch_version: "2.4", cuda_version: "12.1.0" }
50- - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0", latest: true }
50+ - { python_version: "3.12", pytorch_version: "2.5", cuda_version: "12.1.0", tag: "--add_latest" }
5151 steps :
5252 - uses : actions/checkout@v4
5353 with :
@@ -67,11 +67,10 @@ jobs:
6767 run : |
6868 pip install -q -r .actions/requirements.txt
6969 tags = $(python .actions/assistant.py generate_docker_tags \
70- --release-version="${{ env.RELEASE_VERSION }}" \
71- --python-version="${{ matrix.python_version }}" \
72- --torch-version="${{ matrix.pytorch_version }}" \
73- --cuda-version="${{ matrix.cuda_version }}" \
74- --add_latest="${{ matrix.latest }}")
70+ --release_version="${{ env.RELEASE_VERSION }}" \
71+ --python_version="${{ matrix.python_version }}" \
72+ --torch_version="${{ matrix.pytorch_version }}" \
73+ --cuda_version="${{ matrix.cuda_version }}" ${{ matrix.tag }} )
7574 echo "DOCKER_TAGS=$tags" >> $GITHUB_ENV
7675
7776 - uses : docker/build-push-action@v6
0 commit comments