try to fix env to workflow

.github/workflows/build-and_publish-prebuilts-whl.yml

@@ -30,13 +30,10 @@ jobs:
 
       # Step 3: Build the Docker image
       - name: Build Docker image
-        env:
-          ROCM_VERSION: ${{ inputs.rocm_version }}
-          TORCH_VERSION: ${{ inputs.torch_version }}
         run: |
           docker build \
-            --build-arg ROCM_VERSION=${ROCM_VERSION} \
-            --build-arg TORCH_VERSION=${TORCH_VERSION} \
+            --build-arg ROCM_VERSION=${{ inputs.rocm_version }} \
+            --build-arg TORCH_VERSION=${{ inputs.torch_version }} \
             -t prebuilt-wheels ./prebuilts
 
       # Step 4: Create a container and run the script

prebuilts/Dockerfile

@@ -1,8 +1,3 @@
-ARG ROCM_VERSION
-ARG TORCH_VERSION
-ENV ROCM_VERSION=${ROCM_VERSION}
-ENV TORCH_VERSION=${TORCH_VERSION}
-
 FROM rocm/dev-ubuntu-22.04:${ROCM_VERSION}
 
 ENV DEBIAN_FRONTEND=noninteractive \