zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-08-12 04:06:39 +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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats 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: if build_stage == 1:
install_requires += [ install_requires += [
f'mlx-metal=={version}; platform_system == "Darwin"', 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"] = [ extras["cpu"] = [
f'mlx-cpu=={version}; extra == "cpu" and platform_system == "linux"' f'mlx-cpu=={version}; extra == "cpu" and platform_system == "Linux"'
] ]
_setup( _setup(