f5-tts docker workflow - try free diskspace action

.github/workflows/docker-publish-f5-tts.yml

@@ -31,6 +31,17 @@ jobs:
           username: ${{ github.actor }}
           password: ${{ secrets.GITHUB_TOKEN }}
 
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          tool-cache: false
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: false
+          docker-images: false
+          swap-storage: false
+
       - name: Build image
         run: |
           echo "check available disk space"
@@ -39,8 +50,6 @@ jobs:
 
       - name: Push image to GitHub Container Registry
         run: |
-          # testing
-          exit 0
           IMAGE_ID=ghcr.io/${{ github.repository_owner }}/$IMAGE_NAME
 
           # Change all uppercase to lowercase

tts/F5-TTS/Dockerfile

@@ -1,5 +1,5 @@
 # Base CUDA image
-FROM pytorch/pytorch:2.5.0-cuda12.1-cudnn9-runtime
+FROM cnstark/pytorch:2.3.1-py3.10.15-cuda12.1.0-devel-ubuntu22.04
 
 # Set environment variables
 ENV DEBIAN_FRONTEND=noninteractive