From 354348f857265ea97e2ffa6a22dabe0b675c8a89 Mon Sep 17 00:00:00 2001
From: YAN Wenkun <code@yanwk.fun>
Date: Sun, 8 Dec 2024 01:59:49 +0800
Subject: [PATCH] pt241

---
 pt24-cu121/requirements3.txt | 8 ++++----
 pt24-cu121/requirements5.txt | 1 +
 pt24-cu121/requirements9.txt | 4 ----
 pt24-cu121/stage1.sh         | 2 +-
 4 files changed, 6 insertions(+), 9 deletions(-)

diff --git a/pt24-cu121/requirements3.txt b/pt24-cu121/requirements3.txt
index a282629..9970db5 100644
--- a/pt24-cu121/requirements3.txt
+++ b/pt24-cu121/requirements3.txt
@@ -1,6 +1,6 @@
 --index-url https://download.pytorch.org/whl/cu121
 --extra-index-url https://pypi.org/simple
-xformers==0.0.27.post2
-torch==2.4.0
-torchvision==0.19.0
-torchaudio==2.4.0
+xformers==0.0.28.post1
+torch==2.4.1
+torchvision==0.19.1
+torchaudio==2.4.1
diff --git a/pt24-cu121/requirements5.txt b/pt24-cu121/requirements5.txt
index 682e57f..cab1ca9 100644
--- a/pt24-cu121/requirements5.txt
+++ b/pt24-cu121/requirements5.txt
@@ -1,6 +1,7 @@
 # Generated by generate-requirements5.sh
 # Note: Better move versioned items to later steps
 cachetools
+clip-interrogator
 color-matcher
 dill
 gitpython
diff --git a/pt24-cu121/requirements9.txt b/pt24-cu121/requirements9.txt
index bb75a5d..89b9c9e 100644
--- a/pt24-cu121/requirements9.txt
+++ b/pt24-cu121/requirements9.txt
@@ -6,7 +6,3 @@ open-clip-torch==2.24.0
 
 # For CRM V3 (eden_comfy_pipelines)
 # open-clip-torch==2.26.1
-
-# Forcing version to prevent PIP re-installing to latest
-clip-interrogator==0.6.0
-torchvision==0.19.0
diff --git a/pt24-cu121/stage1.sh b/pt24-cu121/stage1.sh
index 40e0347..1ae5f4f 100644
--- a/pt24-cu121/stage1.sh
+++ b/pt24-cu121/stage1.sh
@@ -62,7 +62,7 @@ $pip_exe install -r "$workdir"/requirements6.txt
 $pip_exe install "$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
-$pip_exe install torch-scatter -f https://data.pyg.org/whl/torch-2.4.0%2Bcu121.html
+$pip_exe install torch-scatter -f https://data.pyg.org/whl/torch-2.4.1%2Bcu121.html
 
 $pip_exe install -r "$workdir"/requirements9.txt