added setup py
This commit is contained in:
@@ -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,
|
||||
)
|
||||
Reference in New Issue
Block a user