Upgrade to PyTorch 2.4.0

README.zh.adoc

@@ -25,7 +25,7 @@ https://github.com/ltdrdata/ComfyUI-Impact-Pack[Impact Pack]
 * 包含 240+ 版本兼容的 Python 包，包括：
 ** 安装时经常“找不到编译器”的 `insightface`、`dlib`
 ** 需要从 git 仓库编译安装的 https://github.com/openai/CLIP[CLIP]
-** 运行时经常“找不到 CUDA provider”的 ONNX Runtime，并支持 CUDA 12.1
+** 运行时经常“找不到 CUDA provider”的 ONNX Runtime，并支持 CUDA 12.x
 ** 官方包里没带的 xFormers，细节参见 <<xformers, 关于 xFormers>>.
 
 * 仅预置了少量会卡程序启动的功能性模型，绘图模型还请用户自行下载。

requirements.txt

@@ -10,6 +10,7 @@ compel
 cupy-cuda12x
 deepdiff
 diffusers
+dill
 einops
 fairscale
 filelock
@@ -25,7 +26,7 @@ jaxtyping
 joblib
 kornia
 lark
-matplotlib==3.8
+matplotlib
 matrix-client==0.4.0
 mediapipe
 mss
@@ -42,6 +43,7 @@ opencv-python
 opencv-python-headless
 packaging
 pandas
+peft
 piexif
 pilgram
 pillow

requirements2.txt

@@ -1,4 +1,4 @@
-https://raw.githubusercontent.com/z-mahmud22/Dlib_Windows_Python3.x/main/dlib-19.24.1-cp311-cp311-win_amd64.whl
+https://github.com/eddiehe99/dlib-whl/releases/download/v19.24.5/dlib-19.24.5-cp311-cp311-win_amd64.whl
 https://raw.githubusercontent.com/Gourieff/Assets/main/Insightface/insightface-0.7.3-cp311-cp311-win_amd64.whl
 gradio
 gradio-client

stage1.sh

@@ -37,7 +37,7 @@ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
     --upgrade pip wheel setuptools Cython cmake
 
 ./python.exe -s -m pip install \
-    xformers==0.0.26.post1 torchvision==0.18.0 torchaudio \
+    xformers==0.0.27.post2 torchvision==0.19.0 torchaudio \
     --index-url https://download.pytorch.org/whl/cu121 \
     --extra-index-url https://pypi.org/simple
 
@@ -58,11 +58,11 @@ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
 
 # ComfyUI-3D-Pack, part 2/2
 ./python.exe -s -m pip install \
-    "$workdir"/Comfy3D_Pre_Builds/_Build_Wheels/_Wheels_win_py311_cu121/*.whl
+    "$workdir"/Comfy3D_Pre_Builds/_Build_Wheels/_Wheels_win_py311_torch2.4.0_cu121/*.whl
 
 # From: https://github.com/rusty1s/pytorch_scatter?tab=readme-ov-file#binaries
 ./python.exe -s -m pip install \
-    torch-scatter -f https://data.pyg.org/whl/torch-2.3.0+cu121.html
+    torch-scatter -f https://data.pyg.org/whl/torch-2.4.0%2Bcu121.html
 
 rm -rf "$workdir"/Comfy3D_Pre_Builds