init test

.gitea/workflows/build.yaml

@@ -3,35 +3,10 @@ name: Docker Build and Release Upload
 on:
   push:
     branches:
-      - main
+      - torchaudio-builder
 
 jobs:
 
-  # Build ComfyUI-Cuda
-  build-cuda:
-    runs-on: alpine-linux
-    steps:
-      - name: checkout
-        uses: actions/checkout@v4
-
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-
-      - name: Log in to DockerHub
-        uses: docker/login-action@v3
-        with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_TOKEN }}
-
-      - name: build and push comfyui-cuda
-        run: |
-          docker buildx build \
-          --platform=linux/arm64 \
-          -t mistox/comfyui-cuda-ampere:latest \
-          -f comfyui-cuda-dockerfile \
-          --push \
-          .
-
   # Build ComfyUI-Rocm
   build-rocm:
     runs-on: alpine-linux
@@ -56,28 +31,3 @@ jobs:
           -f comfyui-rocm-dockerfile \
           --push \
           .
-
-  # Job 3: Uvicorn Build
-  build-uvicorn:
-    runs-on: alpine-linux
-    steps:
-      - name: checkout
-        uses: actions/checkout@v4
-
-      - name: Set up Docker Buildx
-        uses: docker/setup-buildx-action@v3
-
-      - name: Log in to DockerHub
-        uses: docker/login-action@v3
-        with:
-          username: ${{ secrets.DOCKERHUB_USERNAME }}
-          password: ${{ secrets.DOCKERHUB_TOKEN }}
-
-      - name: build and push uvicorn
-        run: |
-          docker buildx build \
-          --platform=linux/arm64 \
-          -t mistox/uvicorn-ampere:latest \
-          -f uvicorn-dockerfile \
-          --push \
-          .

comfyui-rocm-dockerfile

@@ -1,21 +1,44 @@
 # Use the official ROCm base image
-FROM ubuntu:26.04
+FROM ubuntu:26.04 as reqs
 
 # Set environment variables for 7900 XTX (gfx1100)
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV ROCM_PATH=/opt/rocm
+ENV PYTORCH_ROCM_ARCH=gfx1100
+ENV HIP_VISIBLE_DEVICES=0
+ENV PYTORCH_ROCM_ARCH=your_gfx_arch
 
 RUN apt update -y && apt install -y --no-install-recommends \
     rocm \
     python3-pip \
     python3-dev \
     python3-torch-rocm \
-    python3-torchaudio \
     git \
     libgl1 \
     libglib2.0-bin \
+    cmake \
+    ninja-build \
+    pkg-config \
+    libavcodec-dev \
+    libavformat-dev \
+    libswresample-dev \
     && rm -rf /var/lib/apt/lists/*
 
+FROM reqs as torch-audio
+
+WORKDIR /builder
+
+RUN git clone https://github.com/pytorch/audio.git && \
+    cd audio && \
+    git submodule update --init --recursive && \
+    # We must ensure it uses the system python/torch
+    BUILD_ROCM=1 python3 setup.py install && \
+    cd .. && rm -rf audio
+
+FROM torch-audio as runner
+
 WORKDIR /app
 
 # Clone ComfyUI