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

@@ -23,7 +23,7 @@ sys.path.append(glob_path)
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()