19 lines
739 B
Docker
19 lines
739 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 echo '#!/bin/bash\ngit status || git clone https://github.com/comfyanonymous/ComfyUI.git /app \npip install -r /app/requirements.txt \npython /app/main.py "$@"' | tee /bin/startup.sh
|
|
|
|
# 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" ]
|