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 = { modules_to_check = {
"torch": ( # really need a better way to check the versions. kinda reinventing the wheel here "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.11.0",
"1.13.1", "1.13.1",
"2.0.0", "2.0.0",
@ -38,6 +40,8 @@ modules_to_check = {
"2.5.1", "2.5.1",
), ),
"torchvision": ( "torchvision": (
"0.11.2",
"0.11.3",
"0.12.0", "0.12.0",
"0.14.1", "0.14.1",
"0.15.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}+cu124" for v in module_versions)
allowed_versions += tuple(f"{v}+cu126" 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.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}+rocm4.5.2" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm5.0" for v in module_versions) allowed_versions += tuple(f"{v}+rocm5.0" for v in module_versions)