From 3c570d38ef4f8989a9a7eeadc4dd2532f6c43daf Mon Sep 17 00:00:00 2001
From: Madhukar Mishra <madhukar93@gmail.com>
Date: Mon, 21 Oct 2024 19:28:13 +0530
Subject: [PATCH] f5-tts docker workflow - try free diskspace action

---
 .github/workflows/docker-publish-f5-tts.yml | 13 +++++++++++--
 tts/F5-TTS/Dockerfile                       |  2 +-
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/docker-publish-f5-tts.yml b/.github/workflows/docker-publish-f5-tts.yml
index 24e60aa..aa3d531 100644
--- a/.github/workflows/docker-publish-f5-tts.yml
+++ b/.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
diff --git a/tts/F5-TTS/Dockerfile b/tts/F5-TTS/Dockerfile
index 2a0a431..4c07faf 100644
--- a/tts/F5-TTS/Dockerfile
+++ b/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