improve: pip handling

- prevent downgrade `torch, torchsde, torchvision, transformers, safetensors, kornia`
This commit is contained in:
dr.lt.data
2024-03-14 12:45:18 +09:00
parent 0dfa84b368
commit d829220973
2 changed files with 37 additions and 6 deletions

View File

@@ -15,6 +15,9 @@ sys.path.append(glob_path)
import cm_global
cm_global.pip_downgrade_blacklist = ['torch', 'torchsde', 'torchvision', 'transformers', 'safetensors', 'kornia']
def skip_pip_spam(x):
return 'Requirement already satisfied:' in x
@@ -350,7 +353,12 @@ def is_installed(name):
if match:
name = match.group(1)
if name in cm_global.pip_downgrade_blacklist:
if match is None or match.group(2) in ['<=', '==', '<']:
print(f"[ComfyUI-Manager] skip black listed pip installation: '{name}'")
return True
return name.lower() in get_installed_packages()