feat: complete pip freezing for torch, torchsde, torchvision

2024-08-16 00:07:19 +09:00
parent d3654b2ee4
commit 61efd60681
4 changed files with 13 additions and 2 deletions
--- a/cm-cli.py
+++ b/cm-cli.py
@@ -35,6 +35,7 @@ restore_snapshot_path = os.path.join(startup_script_path, "restore-snapshot.json
 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):