Fix docker-compose and Dockerfile

.dockerignore

@@ -0,0 +1,3 @@
+checkpoints/
+outputs/
+logs/

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.6.0-runtime-ubuntu22.04 AS base
+FROM nvidia/cuda:12.8.0-runtime-ubuntu22.04 AS base
 
 # Set environment variables
 ENV PYTHONDONTWRITEBYTECODE=1 \
@@ -14,6 +14,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3-venv \
     python3-dev \
     build-essential \
+    ffmpeg \
     git \
     curl \
     wget \
@@ -25,30 +26,19 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
 RUN python -m venv /opt/venv
 ENV PATH="/opt/venv/bin:$PATH"
 
-# Create a non-root user to run the application
-RUN useradd -m -u 1001 appuser
-
 # Set working directory
 WORKDIR /app
 
-# Clone the repository
-RUN git clone https://github.com/ace-step/ACE-Step.git .
+COPY . .
 
 # Install specific PyTorch version compatible with CUDA 12.6
 RUN pip3 install --no-cache-dir --upgrade pip && \
     pip3 install --no-cache-dir hf_transfer peft && \
-    pip3 install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu126
+    pip3 install --no-cache-dir -r requirements.txt --extra-index-url https://download.pytorch.org/whl/cu128
 RUN pip3 install --no-cache-dir .
 
-# Ensure target directories for volumes exist and have correct initial ownership
-RUN mkdir -p /app/outputs /app/checkpoints /app/logs && \
-    chown -R appuser:appuser /app/outputs /app/checkpoints /app/logs
-
-# Change ownership of app files to appuser
-RUN chown -R appuser:appuser /app
+RUN mkdir -p /app/outputs /app/checkpoints /app/logs
 
-# Switch to non-root user
-USER appuser
 
 # Expose the port the app runs on
 EXPOSE 7865
@@ -57,7 +47,7 @@ VOLUME [ "/app/checkpoints", "/app/outputs", "/app/logs" ]
 
 # Set healthcheck
 HEALTHCHECK --interval=60s --timeout=10s --start-period=5s --retries=5 \
-  CMD curl -f http://localhost:7865/ || exit 1
+    CMD curl -f http://localhost:7865/ || exit 1
 
 # Command to run the application with GPU support
-CMD ["python3", "acestep/gui.py", "--server_name", "0.0.0.0", "--bf16", "true"]
+CMD ["python3", "acestep/gui.py", "--server_name", "0.0.0.0", "--bf16", "true"]

docker-compose.yaml

@@ -2,7 +2,7 @@
 services:
   &name ace-step:
     build:
-      context: https://github.com/ace-step/ACE-Step.git
+      context: . 
       dockerfile: Dockerfile
     container_name: *name
     hostname: *name

requirements.txt

@@ -10,6 +10,7 @@ pytorch_lightning==2.5.1
 soundfile==0.13.1
 torch
 torchaudio
+torchcodec
 torchvision
 tqdm
 transformers==4.50.0