try to fix env to workflow

This commit is contained in:
Mathieu Broillet 2024-08-28 15:29:51 +02:00
parent 12b41b2657
commit 1435cf6775
Signed by: mathieu
GPG Key ID: A08E484FE95074C1

View File

@ -31,6 +31,9 @@ jobs:
# Step 3: Build the Docker image
- name: Build Docker image
run: |
echo ${{ inputs.rocm_version }}
echo ${{ inputs.torch_version }}
docker build \
--build-arg ROCM_VERSION=${{ inputs.rocm_version }} \
--build-arg TORCH_VERSION=${{ inputs.torch_version }} \