Eliminate Duplicates

Author: WANDY666

.github/workflows/docker-publish.yml

@@ -91,28 +91,45 @@ jobs:
         with:
           images: ${{ env.REGISTRY }}/${{ env.IMAGE_NAME }}
 
-      # Build and export wheels directly (fastest method)
-      - name: Build and export wheels
+      # Build and export locally first (fast, includes wheels)
+      - name: Build and export files
         uses: docker/build-push-action@v5
         with:
           context: .
-          target: wheels  # We'll add this target to Dockerfile
-          outputs: type=local,dest=./out
+          target: main
+          outputs: type=local,dest=./build-output
           cache-from: type=local,src=/tmp/.buildx-cache
           cache-to: type=local,dest=/tmp/.buildx-cache,mode=max
-
-      # Build and push the complete image (main stage)
-      - name: Build and push Docker image  
+      
+      # Push image to registry (using cache, very fast)
+      - name: Push Docker image
+        if: ${{ github.event_name != 'pull_request' }}
         id: build-and-push
         uses: docker/build-push-action@v5
         with:
           context: .
-          target: main  # 构建主阶段（完整可用镜像）
-          push: ${{ github.event_name != 'pull_request' }}
+          target: main
+          push: true
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
           cache-from: type=local,src=/tmp/.buildx-cache
-          cache-to: type=local,dest=/tmp/.buildx-cache,mode=max
+          # No cache-to here, already cached in first build
+      
+      # Extract wheels from local export (no docker pull needed!)
+      - name: Extract wheels
+        run: |
+          echo "📦 Extracting wheels from build output..."
+          mkdir -p ./out
+          if [ -d "./build-output/out" ]; then
+            cp ./build-output/out/*.whl ./out/
+            echo "✅ Wheels extracted from build output:"
+            ls -la ./out/*.whl
+            echo "Total wheels: $(ls -1 ./out/*.whl | wc -l)"
+          else
+            echo "❌ Warning: /out directory not found in build output"
+            echo "Available directories:"
+            ls -la ./build-output/
+          fi
 
       - name: Sign the published Docker image
         if: ${{ github.event_name != 'pull_request' }}
@@ -121,12 +138,6 @@ jobs:
           DIGEST: ${{ steps.build-and-push.outputs.digest }}
         run: echo "${TAGS}" | xargs -I {} cosign sign --yes {}@${DIGEST}
 
-      - name: Verify exported wheels
-        run: |
-          echo "Wheel files exported directly during build:"
-          ls -la ./out/
-          echo "Wheel count: $(ls -1 ./out/*.whl | wc -l)"
-
       - name: Upload .whl to GitHub Release
         if: startsWith(github.ref, 'refs/tags/')
         env:

Dockerfile

@@ -105,8 +105,4 @@ RUN echo "📦 Using prebuilt flash_attn_3 wheel..." && \
 
 # Install the built wheels (ready-to-use image)
 RUN pip install /out/*.whl && \
-    echo "✅ All wheels installed and ready to use"
-
-# Export stage for GitHub Actions (allows direct wheel extraction)  
-FROM scratch as wheels
-COPY --from=main /out/*.whl /
+    echo "✅ All wheels installed and ready to use"