Getting the setup script to work on Intel based Mac machines

This commit is contained in:
willschneider15
2024-10-22 02:42:16 -04:00
parent 82a708f974
commit e1ae6d5a4f
2 changed files with 13 additions and 8 deletions

View File

@@ -27,13 +27,6 @@ install_requires = [
"tinygrad @ git+https://github.com/tinygrad/tinygrad.git@232edcfd4f8b388807c64fb1817a7668ce27cbad",
]
# Add macOS-specific packages if on Darwin (macOS)
if sys.platform.startswith("darwin"):
install_requires.extend([
"mlx==0.18.0",
"mlx-lm==0.18.2",
])
extras_require = {
"linting": [
"pylint==3.2.6",
@@ -41,6 +34,10 @@ extras_require = {
"mypy==1.11.0",
"yapf==0.40.2",
],
"apple_silicon": [
"mlx==0.18.0",
"mlx-lm==0.18.2",
],
}
setup(