extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 00:03:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #1604 from flavioislima/fix/rocm_url

Browse Source 
FIX: ROCM download URL
This commit is contained in:
cmdr2 2023-09-25 14:06:45 +05:30 committed by GitHub
commit 91e3bfe58f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/check_modules.py
View File

@ -56,14 +56,14 @@ def install(module_name: str, module_version: str):
if "Navi 3" in amd_gpus: if "Navi 3" in amd_gpus:
# No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build # No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build
module_version = "2.1.0.dev-20230614+rocm5.5" module_version = "2.1.0.dev-20230614+rocm5.5"
index_url = "https: //download.pytorch.org/whl/nightly/rocm5.5" index_url = "https://download.pytorch.org/whl/nightly/rocm5.5"
else: else:
module_version = "1.13.1+rocm5.2" module_version = "1.13.1+rocm5.2"
elif module_name == "torchvision": elif module_name == "torchvision":
if "Navi 3" in amd_gpus: if "Navi 3" in amd_gpus:
# No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build # No AMD 7x00 support in rocm 5.2, needs nightly 5.5. build
module_version = "0.16.0.dev-20230614+rocm5.5" module_version = "0.16.0.dev-20230614+rocm5.5"
index_url = "https: //download.pytorch.org/whl/nightly/rocm5.5" index_url = "https://download.pytorch.org/whl/nightly/rocm5.5"
else: else:
module_version = "0.14.1+rocm5.2" module_version = "0.14.1+rocm5.2"
elif os_name == "Darwin": elif os_name == "Darwin":