From 7c12e2da55425b14b6d9705f93528e50ff00d41c Mon Sep 17 00:00:00 2001 From: Madiator2011 Date: Wed, 10 Jul 2024 17:06:13 +0200 Subject: [PATCH] Added xformer to ComfyUI --- official-templates/better-comfyui/Dockerfile | 3 ++- official-templates/better-comfyui/docker-bake.hcl | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/official-templates/better-comfyui/Dockerfile b/official-templates/better-comfyui/Dockerfile index 0bb1ca1..e72bbe5 100644 --- a/official-templates/better-comfyui/Dockerfile +++ b/official-templates/better-comfyui/Dockerfile @@ -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 diff --git a/official-templates/better-comfyui/docker-bake.hcl b/official-templates/better-comfyui/docker-bake.hcl index 1666161..445371d 100644 --- a/official-templates/better-comfyui/docker-bake.hcl +++ b/official-templates/better-comfyui/docker-bake.hcl @@ -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"]