fix: longer timeouts, fix on hf model pull

remsky · Jan 2, 2025 · 3ca79c6 · 3ca79c6
1 parent 7d7da3c
commit 3ca79c6
Show file tree

Hide file tree

Showing 3 changed files with 20 additions and 10 deletions.
diff --git a/.github/workflows/docker-publish.yml b/.github/workflows/docker-publish.yml
@@ -72,7 +72,6 @@ jobs:
           tags: ${{ steps.meta-gpu.outputs.tags }}
           labels: ${{ steps.meta-gpu.outputs.labels }}
           platforms: linux/amd64
-          target: runtime
 
       # Build and push CPU version
       - name: Build and push CPU Docker image
@@ -84,7 +83,6 @@ jobs:
           tags: ${{ steps.meta-cpu.outputs.tags }}
           labels: ${{ steps.meta-cpu.outputs.labels }}
           platforms: linux/amd64
-          target: runtime
 
   create-release:
     needs: build

diff --git a/docker-compose.cpu.yml b/docker-compose.cpu.yml
@@ -8,7 +8,7 @@ services:
       sh -c "
         rm -f .git/index.lock;
         if [ -z \"$(ls -A .)\" ]; then
-          git clone https://huggingface.co/hexgrad/Kokoro-82M
+          git clone https://huggingface.co/hexgrad/Kokoro-82M .
           touch .cloned;
         else
           rm -f .git/index.lock && \
@@ -20,9 +20,9 @@ services:
       "
     healthcheck:
       test: ["CMD", "test", "-f", ".cloned"]
-      interval: 3s
-      timeout: 1s
-      retries: 120
+      interval: 5s
+      timeout: 2s
+      retries: 300
       start_period: 1s
 
   kokoro-tts:
@@ -39,3 +39,15 @@ services:
     depends_on:
       model-fetcher:
         condition: service_healthy
+
+  # Gradio UI service [Comment out everything below if you don't need it]
+  gradio-ui:
+    build:
+      context: ./ui
+    ports:
+      - "7860:7860"
+    volumes:
+      - ./ui/data:/app/ui/data
+      - ./ui/app.py:/app/app.py  # Mount app.py for hot reload
+    environment:
+      - GRADIO_WATCH=True  # Enable hot reloading
diff --git a/docker-compose.yml b/docker-compose.yml
@@ -8,7 +8,7 @@ services:
       sh -c "
         rm -f .git/index.lock;
         if [ -z \"$(ls -A .)\" ]; then
-          git clone https://huggingface.co/hexgrad/Kokoro-82M
+          git clone https://huggingface.co/hexgrad/Kokoro-82M .
           touch .cloned;
         else
           rm -f .git/index.lock && \
@@ -20,9 +20,9 @@ services:
       "
     healthcheck:
       test: ["CMD", "test", "-f", ".cloned"]
-      interval: 3s
-      timeout: 1s
-      retries: 120
+      interval: 5s
+      timeout: 2s
+      retries: 300
       start_period: 1s
 
   kokoro-tts: