extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-05-16 14:24:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

Even older rocm versions

Browse Source
This commit is contained in:
cmdr2 2025-01-09 16:39:21 +05:30
parent 00fc1a81e0
commit be7adcc367
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
scripts/check_modules.py
View File

@ -21,6 +21,8 @@ os_name = platform.system()
modules_to_check = {
"torch": ( # really need a better way to check the versions. kinda reinventing the wheel here
"1.10.1",
"1.10.2",
"1.11.0",
"1.13.1",
"2.0.0",
@ -38,6 +40,8 @@ modules_to_check = {
"2.5.1",
),
"torchvision": (
"0.11.2",
"0.11.3",
"0.12.0",
"0.14.1",
"0.15.1",
@ -302,6 +306,7 @@ def include_cuda_versions(module_versions: tuple) -> tuple:
allowed_versions += tuple(f"{v}+cu124" for v in module_versions)
allowed_versions += tuple(f"{v}+cu126" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm4.2" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm4.3.1" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm4.5.2" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm5.0" for v in module_versions)