diff --git a/.github/workflows/publish-docker.yml b/.github/workflows/publish-docker.yml
index 6f5a605..bdb66b0 100644
--- a/.github/workflows/publish-docker.yml
+++ b/.github/workflows/publish-docker.yml
@@ -10,18 +10,6 @@ on:
 jobs:
   build_and_push:
     runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        include:
-          - image_name: "python:3.11-slim"
-            gpu_type: "other"
-            platforms: "linux/amd64,linux/arm64/v8"
-            tag_name: "other"
-          - image_name: "nvidia/cuda:12.1.1-devel-ubuntu22.04"
-            gpu_type: "nvidia"
-            platforms: "linux/amd64"
-            tag_name: "cuda"
-
     steps:
       - name: Checkout code
         uses: actions/checkout@v4
@@ -41,11 +29,8 @@ jobs:
         with:
           context: .
           file: ./Dockerfile
-          platforms: ${{ matrix.platforms }}
+          platforms: "linux/amd64"
           push: true
           tags: |
-            ghcr.io/josh-xt/alltalk_tts:${{ matrix.tag_name }}
-            ghcr.io/josh-xt/alltalk_tts:${{ matrix.tag_name }}-${{ github.sha }}
-          build-args: |
-            image_name=${{ matrix.image_name }}
-            gpu_type=${{ matrix.gpu_type }}
+            ghcr.io/josh-xt/alltalk_tts:latest
+            ghcr.io/josh-xt/alltalk_tts:main-${{ github.sha }}
diff --git a/Dockerfile b/Dockerfile
index 5a3a349..702953c 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,9 +1,8 @@
-ARG image_name=python:3.11-slim
-FROM ${image_name}
+FROM nvidia/cuda:12.1.1-devel-ubuntu22.04
 WORKDIR /app
 COPY . .
-# Argument for specifying the GPU type, defaulting to "other"
-ARG gpu_type=other
-RUN pip install --no-cache-dir -r requirements_${gpu_type}.txt
+RUN pip install --no-cache-dir -r requirements_nvidia.txt && \
+    pip install --no-cache-dir -r requirements_finetune.txt && \
+    pip install --no-cache-dir -r requirements_other.txt
 EXPOSE 7851
 CMD ["python", "script.py"]