zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-01 04:24:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix install tags (#2464)

Browse Source
This commit is contained in:
Awni Hannun
2025-08-04 20:01:23 -07:00
committed by GitHub
parent 828c5f1137
commit ca973d1e83
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
setup.py
View File

@@ -276,10 +276,10 @@ if __name__ == "__main__":
if build_stage == 1:
install_requires += [
f'mlx-metal=={version}; platform_system == "Darwin"',
f'mlx-cuda=={version}; extra != "cpu" and platform_system == "linux"',
f'mlx-cuda=={version}; extra != "cpu" and platform_system == "Linux"',
]
extras["cpu"] = [
f'mlx-cpu=={version}; extra == "cpu" and platform_system == "linux"'
f'mlx-cpu=={version}; extra == "cpu" and platform_system == "Linux"'
]
_setup(