feat: complete pip freezing for torch, torchsde, torchvision

This commit is contained in:
Dr.Lt.Data
2024-08-16 00:07:19 +09:00
parent d3654b2ee4
commit 61efd60681
4 changed files with 13 additions and 2 deletions

View File

@@ -20,6 +20,7 @@ import cm_global
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()