Madiator2011/madiator-docker-runpod
1
0
Fork 0
mirror of https://github.com/kodxana/madiator-docker-runpod.git synced 2024-12-18 03:56:33 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Added xformer to ComfyUI

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

3
official-templates/better-comfyui/Dockerfile
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

2
official-templates/better-comfyui/docker-bake.hcl
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"]