added setup py

bfaeccc7d528a3ac9ee75bff11e4e37f92582055 · 2024-07-17 13:21:46 +0200 · Alex

Files touched

A setup.py

Diff

commit bfaeccc7d528a3ac9ee75bff11e4e37f92582055
Author: Alex <46456279+the-alex-b@users.noreply.github.com>
Date:   Wed Jul 17 13:21:46 2024 +0200

    added setup py
---
 setup.py | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/setup.py b/setup.py
new file mode 100644
index 00000000..cc07c468
--- /dev/null
+++ b/setup.py
@@ -0,0 +1,40 @@
+from setuptools import setup, find_packages
+import sys
+
+# Base requirements for all platforms
+install_requires = [
+    "aiohttp==3.9.5",
+    "grpcio==1.64.1",
+    "grpcio-tools==1.64.1",
+    "huggingface-hub==0.23.4",
+    "Jinja2==3.1.4",
+    "numpy==2.0.0",
+    "protobuf==5.27.1",
+    "psutil==6.0.0",
+    "pynvml==11.5.3",
+    "requests==2.32.3",
+    "safetensors==0.4.3",
+    "tiktoken==0.7.0",
+    "tokenizers==0.19.1",
+    "tqdm==4.66.4",
+    "transformers==4.41.2",
+    "uuid==1.30",
+    "tinygrad @ git+https://github.com/tinygrad/tinygrad.git@a9f5a764dc640a5e5cbaaeeee21df7c8ca37da38",
+]
+
+# Add macOS-specific packages if on Darwin (macOS)
+if sys.platform == "darwin":
+    install_requires.extend(
+        [
+            "mlx==0.15.1",
+            "mlx-lm==0.14.3",
+        ]
+    )
+
+
+setup(
+    name="exo",
+    version="0.1.0",
+    packages=find_packages(),
+    install_requires=install_requires,
+)

← 12fcbc0d switch over to psutil, more robust system detection · back to Exo · more complete python gitignore template 4ea7db85 →