Added xformer to ComfyUI

This commit is contained in:
Madiator2011 2024-07-10 17:06:13 +02:00 committed by GitHub
parent 98ed2b08da
commit 7c12e2da55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions

View file

@ -21,7 +21,8 @@ RUN git clone https://github.com/comfyanonymous/ComfyUI.git && \
pip install -r requirements.txt && \
git clone https://github.com/ltdrdata/ComfyUI-Manager.git custom_nodes/ComfyUI-Manager && \
cd custom_nodes/ComfyUI-Manager && \
pip install -r requirements.txt
pip install -r requirements.txt && \
pip install -U xformers --index-url https://download.pytorch.org/whl/cu121
# Stage 3: Model Setup and Final Image
FROM comfyui-install as final

View file

@ -11,7 +11,7 @@ target "py310-cuda121" {
dockerfile = "Dockerfile"
args = {
BASE_IMAGE = "nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04",
TORCH = "torch==2.3.1+cu121 -f https://download.pytorch.org/whl/torch_stable.html",
TORCH = "torch==2.3.0+cu121 -f https://download.pytorch.org/whl/torch_stable.html",
PYTHON_VERSION = "3.10"
}
tags = ["madiator2011/better-comfyui:cuda12.1"]