From 5e8e704a4f9c04114b9049a722b7920a6f0a8117 Mon Sep 17 00:00:00 2001
From: YAN Wenkun <code@yanwk.fun>
Date: Wed, 8 Jan 2025 21:00:09 +0800
Subject: [PATCH] Update deps

---
 cu124-cn/Dockerfile-cn              | 8 +++++++-
 cu124-slim/builder-scripts/pak5.txt | 1 +
 rocm/builder-scripts/pak5.txt       | 1 +
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/cu124-cn/Dockerfile-cn b/cu124-cn/Dockerfile-cn
index 222d438..8cde241 100644
--- a/cu124-cn/Dockerfile-cn
+++ b/cu124-cn/Dockerfile-cn
@@ -160,7 +160,13 @@ RUN --mount=type=cache,target=/root/.cache/pip \
 
 RUN --mount=type=cache,target=/root/.cache/pip \
     pip install \
-        -r /builder-scripts/pak7.txt \
+        -r /builder-scripts/pak7.txt
+
+# 确保依赖项匹配 ComfyUI & Manager
+RUN --mount=type=cache,target=/root/.cache/pip \
+    pip install \
+        -r https://gh-proxy.com/https://github.com/comfyanonymous/ComfyUI/raw/refs/heads/master/requirements.txt \
+        -r https://gh-proxy.com/https://github.com/ltdrdata/ComfyUI-Manager/raw/refs/heads/main/requirements.txt \
     && pip list
 
 ################################################################################
diff --git a/cu124-slim/builder-scripts/pak5.txt b/cu124-slim/builder-scripts/pak5.txt
index 26d8fc1..a574828 100644
--- a/cu124-slim/builder-scripts/pak5.txt
+++ b/cu124-slim/builder-scripts/pak5.txt
@@ -40,6 +40,7 @@ soundfile
 spandrel
 svglib
 tokenizers
+toml
 torchsde
 tqdm
 transparent-background
diff --git a/rocm/builder-scripts/pak5.txt b/rocm/builder-scripts/pak5.txt
index 7461dae..7ff9db4 100644
--- a/rocm/builder-scripts/pak5.txt
+++ b/rocm/builder-scripts/pak5.txt
@@ -30,6 +30,7 @@ soundfile
 spandrel
 svglib
 tokenizers
+toml
 torchsde
 tqdm
 transparent-background