Merge branch 'switch-docker-base-image' into moeflow-companion-main

.github/workflows/release.yml

@@ -12,27 +12,28 @@ jobs:
     steps:
       - name: Check out the repo
         uses: actions/checkout@v3
-      
+
       - name: Log in to Docker Hub
         uses: docker/login-action@v2
+        if: github.repository == "zyddnys/manga-image-translator"
         with:
           username: ${{ secrets.DOCKER_USERNAME }}
           password: ${{ secrets.DOCKER_PASSWORD }}
-      
+
       - name: Extract metadata
         id: meta
         uses: docker/metadata-action@v2
         with:
           images: zyddnys/manga-image-translator
-      
+
       - name: Build and push Docker image
         uses: docker/build-push-action@ad44023a93711e3deb337508980b4b5e9bcdc5dc
         with:
           context: .
           push: true
           tags: ${{ steps.meta.outputs.tags }}
           labels: ${{ steps.meta.outputs.labels }}
-      
+
       - name: Docker Hub Description
         uses: peter-evans/dockerhub-description@v3
         with:

Dockerfile

@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:latest
+FROM pytorch/pytorch:2.5.1-cuda11.8-cudnn9-runtime
 
 WORKDIR /app
 
@@ -12,11 +12,6 @@ RUN apt-get install -y git g++ ffmpeg libsm6 libxext6 libvulkan-dev
 COPY requirements.txt /app/requirements.txt
 
 RUN pip install -r /app/requirements.txt
-RUN pip install torchvision --force-reinstall
-RUN pip install "numpy<2.0"
-
-RUN apt-get remove -y g++ && \
-    apt-get autoremove -y
 
 # Copy app
 COPY . /app