From 894b87722b118c01ddc4743cfc35b79feacee40a Mon Sep 17 00:00:00 2001
From: simonmcnair <101189766+simonmcnair@users.noreply.github.com>
Date: Tue, 8 Apr 2025 15:17:55 +0100
Subject: [PATCH] Update Dockerfile

update Pytorch and add comfy manager
---
 services/comfy/Dockerfile | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/services/comfy/Dockerfile b/services/comfy/Dockerfile
index 3984675..5a6203f 100644
--- a/services/comfy/Dockerfile
+++ b/services/comfy/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.3.0-cuda12.1-cudnn8-runtime
+FROM pytorch/pytorch:2.5.1-cuda12.4-cudnn9-runtime
 
 ENV DEBIAN_FRONTEND=noninteractive PIP_PREFER_BINARY=1
 
@@ -51,6 +51,9 @@ RUN --mount=type=cache,target="$USER_HOME"/.cache/pip \
   git reset --hard 276f8fce9f5a80b500947fb5745a4dde9e84622d && \
   pip install -r requirements.txt
 
+RUN git clone -b 3.6.5 --depth=1  https://github.com/ltdrdata/ComfyUI-Manager ${ROOT}/custom_nodes/comfyui-manager && \
+  pip install -r ${ROOT}/custom_nodes/comfyui-manager/requirements.txt
+
 RUN chown -R "$PUID:$PGID" "${ROOT}"
 
 WORKDIR ${ROOT}