wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-15 16:02:32 +08:00
Code Issues Actions 20 Packages Projects Releases Wiki Activity

Add pytorch-triton-rocm to dependencies when targeting AMD because accelerate needs to find it in the rocm repo

Browse Source
This commit is contained in:
doctorpangloss 2024-05-22 21:21:23 -07:00
parent 0fcd07962f
commit c0fc1d1458
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
setup.py
View File

@ -146,6 +146,7 @@ def dependencies(force_nightly: bool = False) -> List[str]:
index_urls += [nvidia_torch_index] index_urls += [nvidia_torch_index]
elif _is_amd(): elif _is_amd():
index_urls += [amd_torch_index] index_urls += [amd_torch_index]
_dependencies += ["pytorch-triton-rocm"]
else: else:
index_urls += [cpu_torch_index] index_urls += [cpu_torch_index]