Use torch 2.0.1 and torchvision 0.15.2 by default on Windows

This commit is contained in:
cmdr2 2023-08-29 18:52:06 +05:30
parent 2f910c69b8
commit fbafa56ecb

View File

@ -16,8 +16,8 @@ import traceback
os_name = platform.system()
modules_to_check = {
"torch": ("1.11.0", "1.13.1", "2.0.0"),
"torchvision": ("0.12.0", "0.14.1", "0.15.1"),
"torch": ("1.11.0", "1.13.1", "2.0.0", "2.0.1"),
"torchvision": ("0.12.0", "0.14.1", "0.15.1", "0.15.2"),
"sdkit": "2.0.3",
"stable-diffusion-sdkit": "2.1.4",
"rich": "12.6.0",