build with no libraries and pass in the device via compose

.gitea/workflows/build.yaml

@@ -23,7 +23,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push comfyui-cuda
         run: |
           docker buildx build \
           --platform=linux/arm64 \
@@ -48,7 +48,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push comfyui-rocm
         run: |
           docker buildx build \
           --platform=linux/arm64 \
@@ -73,7 +73,7 @@ jobs:
           username: ${{ secrets.DOCKERHUB_USERNAME }}
           password: ${{ secrets.DOCKERHUB_TOKEN }}
 
-      - name: build and push comfyui
+      - name: build and push uvicorn
         run: |
           docker buildx build \
           --platform=linux/arm64 \

comfyui-rocm-dockerfile

@@ -1,19 +1,15 @@
 # Use the official ROCm base image
-FROM rocm/dev-ubuntu-24.04:6.2
+FROM ubuntu:24.04
 
 # Set environment variables for 7900 XTX (gfx1100)
 ENV HSA_OVERRIDE_GFX_VERSION=11.0.0
 ENV TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL=1
 ENV PYTHONUNBUFFERED=1
 
-# Install system dependencies
+# Install basic Python and Graphics libs
-RUN apk add --no-cache \
+RUN apt update -y && apt install -y \
-    python3 \
+    python3-pip python3-dev git libgl1-mesa-glx libglib2.0-0 \
-    py3-pip \
+    && rm -rf /var/lib/apt/lists/*
-    git \
-    mesa-gles \
-    glib \
-    && rm -rf /var/cache/apk/*
 
 # Set working directory
 WORKDIR /app