Update README.md

.gitea/workflows/build-comfyui-cuda.yaml

@@ -0,0 +1,41 @@
+name: Docker Build and Release Upload
+
+on:
+  push:
+    branches:
+      - torchaudio-builder
+
+jobs:
+
+  # Build ComfyUI-Rocm
+  build-comfyui-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: Log in to docker.mistox.net
+        uses: docker/login-action@v3
+        with:
+          registry: docker.mistox.net
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: build and push comfyui-cuda
+        run: |
+          docker buildx build \
+          --platform=linux/arm64 \
+          -t docker.mistox.net/comfyui-cuda-ampere:latest \
+          -t mistox/comfyui-cuda-ampere:latest \
+          -f comfyui-cuda-dockerfile \
+          --push \
+          .

.gitea/workflows/build-comfyui-rocm.yaml

@@ -8,7 +8,7 @@ on:
 jobs:
 
   # Build ComfyUI-Rocm
-  build-rocm:
+  build-comfyui-rocm:
     runs-on: alpine-linux
     steps:
       - name: checkout
@@ -23,11 +23,19 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
+      - name: Log in to docker.mistox.net
+        uses: docker/login-action@v3
+        with:
+          registry: docker.mistox.net
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
       - name: build and push rocm-aarch64-base
         run: |
           docker buildx build \
           --platform=linux/arm64 \
           -t docker.mistox.net/rocm-aarch64-base:latest \
+          -t mistox/rocm-aarch64-base:latest \
           -f rocm-base-aarch64-dockerfile \
           --push \
           .
@@ -37,24 +45,7 @@ jobs:
           docker buildx build \
           --platform=linux/arm64 \
           -t docker.mistox.net/comfyui-rocm-ampere:latest \
+          -t mistox/comfyui-rocm-ampere:latest \
           -f comfyui-rocm-dockerfile \
           --push \
-          .
-
-      - name: build and push comfyui-cuda
-        run: |
-          docker buildx build \
-          --platform=linux/arm64 \
-          -t docker.mistox.net/comfyui-cuda-ampere:latest \
-          -f comfyui-cuda-dockerfile \
-          --push \
-          .
-      
-      - name: build and push uvicorn
-        run: |
-          docker buildx build \
-          --platform=linux/arm64 \
-          -t docker.mistox.net/uvicorn-ampere:latest \
-          -f uvicorn-dockerfile \
-          --push \
           .

.gitea/workflows/build-uvicorn.yaml

@@ -0,0 +1,41 @@
+name: Docker Build and Release Upload
+
+on:
+  push:
+    branches:
+      - torchaudio-builder
+
+jobs:
+
+  # Build uvicorn
+  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: Log in to docker.mistox.net
+        uses: docker/login-action@v3
+        with:
+          registry: docker.mistox.net
+          username: ${{ secrets.DOCKERHUB_USERNAME }}
+          password: ${{ secrets.DOCKERHUB_TOKEN }}
+
+      - name: build and push uvicorn
+        run: |
+          docker buildx build \
+          --platform=linux/arm64 \
+          -t docker.mistox.net/uvicorn-ampere:latest \
+          -t mistox/uvicorn-ampere:latest \
+          -f uvicorn-dockerfile \
+          --push \
+          .

README.md

@@ -4,5 +4,8 @@ In order to use this repos please just copy the docker-compose
   
 ONLY TESTED WITH AMPERE ALTRA USING RTX40 SERIES  
   
-https://hub.docker.com/r/mistox/uvicorn-ampere  
+https://hub.docker.com/r/mistox/uvicorn-ampere - working  
-https://hub.docker.com/r/mistox/comfyui-ampere
+https://hub.docker.com/r/mistox/comfyui-cuda-ampere - working  
+  
+https://hub.docker.com/r/mistox/rocm-aarch64-base - work in progress  
+https://hub.docker.com/r/mistox/comfyui-rocm-ampere - pending rocm-aarch64-base before testing  

comfyui-cuda-dockerfile

@@ -1,8 +1,31 @@
 # Get cuda version from https://hub.docker.com/r/nvidia/cuda
-FROM nvcr.io/nvidia/pytorch:25.05-py3 AS build-deps
+FROM nvcr.io/nvidia/pytorch:26.04-py3 AS build-deps
 
-# Install system dependencies
+# Set ENV
-RUN pip install imageio controlnet_aux
+ENV PIP_BREAK_SYSTEM_PACKAGES=1
+ENV PYTHONUNBUFFERED=1
+ENV DEBIAN_FRONTEND=noninteractive
+ENV USE_CUDA=1
+ENV BUILD_SOX=1
+
+# Install system packages to build torchaudio
+RUN apt update && \
+    apt upgrade -y && \
+    apt install -y \
+    libavformat-dev \
+    libavcodec-dev \
+    libavutil-dev \
+    libavdevice-dev \
+    libavfilter-dev \
+    libswresample-dev \
+    sox \
+    libsox-dev
+
+# 2. Clone and build the torchaudio stable
+RUN wget https://github.com/pytorch/audio/archive/refs/tags/v2.11.0.tar.gz && \
+    tar -xf v2.11.0.tar.gz && \
+    cd audio-2.11.0 && \
+    pip install --no-build-isolation --no-deps -v .
 
 # Set working directory
 WORKDIR /app
@@ -10,13 +33,9 @@ WORKDIR /app
 # Clone ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
 
-# Link in python
+# Install depencies
-RUN ln -sf /usr/bin/python3 /usr/bin/python && \
-    ln -sf /usr/bin/pip3 /usr/bin/pip
-
-# Install
 RUN grep -vE "^(torch|#)" requirements.txt | xargs pip install && \
-    pip install torchsde
+    pip install torchsde imageio controlnet_aux
 
 # Set custom Nodes
 WORKDIR /app/custom_nodes

comfyui-rocm-dockerfile

@@ -1,13 +1,27 @@
 # Use the rocm-base-aarch64-dockerfile
-FROM ubuntu:26.04 AS reqs
+FROM mistox/rocm-aarch64-base:latest AS base
+
+# Set Workdir
 WORKDIR /app
+
+# Clone ComfyUI
 RUN git clone https://github.com/comfyanonymous/ComfyUI.git .
-RUN pip3 install -r requirements.txt
+
-# Setup Manager
+# Install depencies
+RUN grep -vE "^(torch|#)" requirements.txt | xargs pip install && \
+    pip install torchsde imageio controlnet_aux
+
+# Set custom Nodes
 WORKDIR /app/custom_nodes
+
+# Install ComfyUi Manager
 RUN git clone https://github.com/Comfy-Org/ComfyUI-Manager.git
-# Setup for run
+
+# Go back to root
 WORKDIR /app
+
+# Expose the default ComfyUI port
 EXPOSE 8188
-# Launch
+
+# Launch ComfyUI
 CMD ["python3", "main.py", "--listen", "0.0.0.0"]

rocm-base-aarch64-dockerfile

@@ -4,7 +4,11 @@ FROM ubuntu:26.04 AS reqs
 ENV PIP_BREAK_SYSTEM_PACKAGES=1
 ENV PYTHONUNBUFFERED=1
 ENV DEBIAN_FRONTEND=noninteractive
-ENV MAX_JOBS=10
+
+# My sad server with 96 Gbytes of ram keeps running out of ram ;(
+# I would love to use the whole 80 core CPU ;(
+ENV MAX_JOBS=5
+
 # Install libraries
 RUN apt update -y && \
     apt upgrade -y && \
@@ -26,9 +30,15 @@ RUN apt update -y && \
     flex \
     cmake \
     build-essential \
+    libsimde-dev \
     zlib1g-dev \
     libncurses5-dev \
     libopenblas-dev \
+    libjpeg-dev \
+    libpng-dev \
+    libavcodec-dev \
+    libavformat-dev \
+    libswscale-dev \
     libpthreadpool-dev \
     libgoogle-glog-dev \
     libgtest-dev \
@@ -67,7 +77,7 @@ RUN wget https://www.python.org/ftp/python/3.11.15/Python-3.11.15.tgz && \
     # Tested Working
 
 # Build ROCm latest
-FROM python AS ROCm
+FROM python AS rocm
 WORKDIR /build/therock
 COPY ./TheRock .
 ENV CFLAGS="-Wno-error=discarded-qualifiers -fcommon -w"
@@ -92,7 +102,7 @@ RUN env INSTALL_PREFIX=/usr/local ./dockerfiles/install_pinned_patchelf.sh && \
     # Work in progress [ Breaking due to x86 intrinsics not ported to Aarch64 ]
 
 # Build pytorch
-FROM ROCm AS pytorch
+FROM rocm AS pytorch
 # Setup Envronment
 ENV PYTORCH_ROCM_ARCH=1100
 ENV USE_ROCM=1
@@ -142,12 +152,22 @@ RUN git clone https://github.com/pytorch/pytorch.git && \
 # Build pytorch-audio
 FROM pytorch AS torchaudio
 WORKDIR /build/torchaudio
+RUN wget https://github.com/pytorch/audio/archive/refs/tags/v2.11.0.tar.gz && \
+    tar -xf v2.11.0.tar.gz && \
+    cd audio-2.11.0 && \
+    pip install --no-build-isolation --no-deps -v .
+    # Tested Working via comfyui-cuda-dockerfile
 
 # Build pytorch-vision
 FROM torchaudio AS torchvision
 WORKDIR /build/torchaudio
-RUN git clone https://github.com/pytorch/vision.git
+ENV FORCE_CUDA=0
+RUN git clone https://github.com/pytorch/vision.git . && \
+    pip install numpy pillow ninja setuptools wheel && \
+    python setup.py bdist_wheel && \
+    pip install dist/torchvision-*.whl
+    # Not tested and not verified ROCM build
 
 # Build final stripped down image
-FROM ubuntu:26.04 as rocm-aarch64
+FROM ubuntu:26.04
 # Copy in rocm, pytorch, torchaudio, torchvision