Update build-docker.yaml

thiner · web-flow · commit 5e90adc83bde · 2024-02-19T17:51:06.000+08:00
diff --git a/.github/workflows/build-docker.yaml b/.github/workflows/build-docker.yaml
@@ -34,13 +34,13 @@ jobs:
         uses: docker/build-push-action@v4
         with:
           context: .
-          file: "docker/simple/Dockerfile"
+          file: "docker/cuda_simple/Dockerfile"
           push: ${{ startsWith(github.ref, 'refs/tags/') }}
           pull: true
           platforms: linux/amd64,linux/arm64
           tags: |
-            ghcr.io/thiner/llama-cpp-python:latest
-            ghcr.io/thiner/llama-cpp-python:${{ github.ref_name }}
+            ghcr.io/thiner/llama-cpp-python:latest-cuda
+            ghcr.io/thiner/llama-cpp-python:${{ github.ref_name }}-cuda
           build-args: |
             BUILDKIT_INLINE_CACHE=1
 
