23 lines
738 B
Docker
23 lines
738 B
Docker
FROM intel/intel-extension-for-pytorch:2.7.10-xpu
|
|
|
|
# Optional, might help with memory allocation performance and scalability
|
|
RUN apt-get update && \
|
|
apt-get install -y --no-install-recommends --fix-missing libjemalloc-dev
|
|
ENV LD_PRELOAD=libjemalloc.so.2
|
|
|
|
# Download the ComfyUI repository
|
|
RUN cat <<EOF > /bin/startup.sh
|
|
#!/bin/bash
|
|
git status || git clone https://github.com/comfyanonymous/ComfyUI.git /app
|
|
pip install -r /app/requirements.txt
|
|
python /app/main.py "\$@"
|
|
EOF
|
|
|
|
# Make the startup script executable
|
|
RUN chmod 755 /bin/startup.sh
|
|
|
|
# Set the working directory to /app
|
|
WORKDIR /app
|
|
|
|
# Run ComfyUI with custom parameters
|
|
CMD [ "startup.sh", "--highvram", "--use-pytorch-cross-attention", "--listen=0.0.0.0", "--port=8188" ]
|