updated: PIPFixer - support for pytorch 2.7.0

This commit is contained in:
Dr.Lt.Data
2025-04-28 23:49:42 +09:00
parent 9e132811bc
commit 5978b6c9ee
3 changed files with 3 additions and 2 deletions

View File

@@ -43,7 +43,7 @@ import manager_downloader
from node_package import InstalledNodePackage
version_code = [3, 31, 12]
version_code = [3, 31, 13]
version_str = f"V{version_code[0]}.{version_code[1]}" + (f'.{version_code[2]}' if len(version_code) > 2 else '')

View File

@@ -307,6 +307,7 @@ def parse_requirement_line(line):
torch_torchvision_torchaudio_version_map = {
'2.7.0': ('0.22.0', '2.7.0'),
'2.6.0': ('0.21.0', '2.6.0'),
'2.5.1': ('0.20.0', '2.5.0'),
'2.5.0': ('0.20.0', '2.5.0'),