feat: complete pip freezing for torch, torchsde, torchvision

Author: ltdrdata

cm-cli.py

@@ -35,6 +35,7 @@
 pip_overrides_path = os.path.join(comfyui_manager_path, "pip_overrides.json")
 git_script_path = os.path.join(comfyui_manager_path, "git_helper.py")
 
+cm_global.pip_blacklist = ['torch', 'torchsde', 'torchvision']
 cm_global.pip_downgrade_blacklist = ['torch', 'torchsde', 'torchvision', 'transformers', 'safetensors', 'kornia']
 cm_global.pip_overrides = {}
 if os.path.exists(pip_overrides_path):

glob/manager_core.py

@@ -23,7 +23,7 @@
 import cm_global
 from manager_util import *
 
-version = [2, 48, 8]
+version = [2, 49]
 version_str = f"V{version[0]}.{version[1]}" + (f'.{version[2]}' if len(version) > 2 else '')
 
 
@@ -103,6 +103,9 @@ def is_blacklisted(name):
     if match:
         name = match.group(1)
 
+    if name in cm_global.pip_blacklist:
+        return True
+
     if name in cm_global.pip_downgrade_blacklist:
         pips = get_installed_packages()
 
@@ -129,6 +132,9 @@ def is_installed(name):
     if match:
         name = match.group(1)
 
+    if name in cm_global.pip_blacklist:
+        return True
+
     if name in cm_global.pip_downgrade_blacklist:
         pips = get_installed_packages()
 

prestartup_script.py

@@ -20,6 +20,7 @@
 
 security_check.security_check()
 
+cm_global.pip_blacklist = ['torch', 'torchsde', 'torchvision']
 cm_global.pip_downgrade_blacklist = ['torch', 'torchsde', 'torchvision', 'transformers', 'safetensors', 'kornia']
 
 
@@ -454,6 +455,9 @@ def is_installed(name):
     if match:
         name = match.group(1)
 
+    if name in cm_global.pip_blacklist:
+        return True
+
     if name in cm_global.pip_downgrade_blacklist:
         pips = get_installed_packages()
 

pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "comfyui-manager"
 description = "ComfyUI-Manager provides features to install and manage custom nodes for ComfyUI, as well as various functionalities to assist with ComfyUI."
-version = "2.48.8"
+version = "2.49"
 license = { file = "LICENSE.txt" }
 dependencies = ["GitPython", "PyGithub", "matrix-client==0.4.0", "transformers", "huggingface-hub>0.20", "typer", "rich", "typing-extensions"]