extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-06 13:39:15 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow older rocm versions

Browse Source
This commit is contained in:
cmdr2 2025-01-09 15:59:28 +05:30
parent 5023619676
commit 00fc1a81e0
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

@ -301,6 +301,11 @@ def include_cuda_versions(module_versions: tuple) -> tuple:
allowed_versions += tuple(f"{v}+cu121" 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}+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)
allowed_versions += tuple(f"{v}+rocm5.1.1" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm5.2" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm5.3" for v in module_versions)
allowed_versions += tuple(f"{v}+rocm5.4.2" for v in module_versions)