wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-17 17:50:51 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

select bog standard for linux arm64 since nightlies aren't built for it

Browse Source
This commit is contained in:
Benjamin Berman 2023-08-04 16:26:13 -07:00
parent 3216d1c6b0
commit 3ed81c19e9
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
setup.py
View File

@ -41,6 +41,12 @@ This includes macOS MPS support.
""" """
cpu_torch_index_nightlies = "https://download.pytorch.org/whl/nightly/cpu" cpu_torch_index_nightlies = "https://download.pytorch.org/whl/nightly/cpu"
"""
The package index to torch built against CPU features.
Non-nightlies are selected when building Linux on arm64.
"""
cpu_torch_index = "https://download.pytorch.org/whl/cpu"
# xformers not required for new torch # xformers not required for new torch
""" """
@ -98,6 +104,13 @@ def _is_amd() -> bool:
return False return False
def _is_linux_arm64():
os_name = platform.system()
architecture = platform.machine()
return os_name == 'Linux' and architecture == 'aarch64'
def dependencies() -> [str]: def dependencies() -> [str]:
_dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines() _dependencies = open(os.path.join(os.path.dirname(__file__), "requirements.txt")).readlines()
# todo: also add all plugin dependencies # todo: also add all plugin dependencies
@ -113,6 +126,8 @@ def dependencies() -> [str]:
elif _is_amd(): elif _is_amd():
index_urls += [amd_torch_index] index_urls += [amd_torch_index]
gpu_accelerated = True gpu_accelerated = True
elif _is_linux_arm64():
index_urls += [cpu_torch_index]
else: else:
index_urls += [cpu_torch_index_nightlies] index_urls += [cpu_torch_index_nightlies]