owner/comfyui-docker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nvidia updates

Browse Source
This commit is contained in:
owner
2025-11-17 02:03:28 -05:00
parent 4d9a7bbee5
commit 8a388f3d92
2 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
services/nvidia/Dockerfile
View File

@@ -40,6 +40,9 @@ RUN pip install --break-system-packages -r requirements.txt \
&& pip install --break-system-packages -r /ComfyUI/custom_nodes/ComfyUI-SaveAs/requirements.txt \
&& pip install --break-system-packages gitpython psutil
# Install prebuilt flash-attn
RUN pip install --break-system-packages https://github.com/mjun0812/flash-attention-prebuild-wheels/releases/download/v0.4.22/flash_attn-2.8.1+cu130torch2.9-cp312-cp312-linux_x86_64.whl
# Set ownership and move models directory
RUN chown -R $DOCKER_USER:$DOCKER_USER /ComfyUI && \
mv /ComfyUI/models /ComfyUI/models.orig

2
services/nvidia/start.sh
View File

@@ -2,4 +2,4 @@
echo "COPYING"
cp -rf /ComfyUI/models.orig/* /ComfyUI/models/
echo "START"
python3 main.py "${CLI_ARGS}" --listen
python3 main.py --listen "${CLI_ARGS}"