migraph fix

Dockerfiles/Dockerfile.comfyui-rocm7.1

@@ -6,10 +6,12 @@ RUN apt-get update \
        libgl1 libglib2.0-dev libsm6 libxext6 libxrender-dev \
        libgomp1 libgstreamer1.0-0 libgstreamer-plugins-base1.0-0 \
        libegl1 libxrandr2 libxss1 libxcursor1 libxcomposite1 libxdamage1 \
-       libxfixes3 libxi6 libxinerama1 libxrandr2 libxrender1 \
+       libxfixes3 libxi6 libxinerama1 libxrandr2 libxrender1 migraphx \
     && apt-get clean \
     && rm -rf /var/lib/apt/lists/*
 
+
+
 # Copy ComfyUI repository and pre-built virtual environment
 WORKDIR /app
 

comfyui-build/docker-entrypoint.sh

@@ -111,6 +111,9 @@ fi
 #pip install --upgrade flash_attn --no-build-isolation
 echo "Installing ComfyUI requirements..."
 pip install -r requirements.txt
+pip install ultralytics
+pip install onnxruntime-rocm
+
 
 echo "Starting ComfyUI..."
 if [ -f "start.sh" ]; then

comfyui-setup.sh

@@ -2,21 +2,26 @@ cd User-Directories/ComfyUI/custom_nodes
 
 # Manager
 git clone https://github.com/ltdrdata/ComfyUI-Manager comfyui-manager
+cd comfyui-manager && pip install -r requirements.txt && cd ..
 
 # Crystools (AMD branch)
 git clone -b AMD https://github.com/crystian/ComfyUI-Crystools.git
+cd ComfyUI-Crystools && pip install -r requirements.txt && cd ..
 
 # MIGraphX
 git clone https://github.com/pnikolic-amd/ComfyUI_MIGraphX.git
+cd ComfyUI_MIGraphX && pip install -r requirements.txt && cd ..
 
 # Unsafe Torch
 git clone https://github.com/ltdrdata/comfyui-unsafe-torch
 
 # Impact Pack
 git clone https://github.com/ltdrdata/ComfyUI-Impact-Pack comfyui-impact-pack
+cd comfyui-impact-pack && pip install -r requirements.txt && cd ..
 
 # Impact Subpack
 git clone https://github.com/ltdrdata/ComfyUI-Impact-Subpack
+cd ComfyUI-Impact-Subpack && pip install -r requirements.txt && cd ..
 
 # WaveSpeed
 git clone https://github.com/chengzeyi/Comfy-WaveSpeed.git

docker-compose.yaml

@@ -133,6 +133,7 @@ services:
       - HSA_ENABLE_SDMA=1
       - MIOPEN_FIND_MODE=2
       - MIOPEN_ENABLE_CACHE=1
+      - MIGRAPHX_MLIR_USE_SPECIFIC_OPS="attention"
     ports:
       - "8188:8188"
     networks: