|
|
|
|
|
""" |
|
|
MiniMind (Mind2) - Setup Script |
|
|
Lightweight language models for edge deployment. |
|
|
""" |
|
|
|
|
|
from setuptools import setup, find_packages |
|
|
from pathlib import Path |
|
|
|
|
|
|
|
|
readme_path = Path(__file__).parent / "README.md" |
|
|
long_description = readme_path.read_text(encoding="utf-8") if readme_path.exists() else "" |
|
|
|
|
|
|
|
|
req_path = Path(__file__).parent / "requirements.txt" |
|
|
requirements = [] |
|
|
if req_path.exists(): |
|
|
requirements = [ |
|
|
line.strip() for line in req_path.read_text().splitlines() |
|
|
if line.strip() and not line.startswith("#") |
|
|
] |
|
|
|
|
|
setup( |
|
|
name="minimind", |
|
|
version="1.0.0", |
|
|
author="Matrix Agent", |
|
|
author_email="contact@minimind.ai", |
|
|
description="MiniMind (Mind2) - Lightweight language models for edge deployment", |
|
|
long_description=long_description, |
|
|
long_description_content_type="text/markdown", |
|
|
url="https://github.com/minimind/minimind", |
|
|
project_urls={ |
|
|
"Documentation": "https://github.com/minimind/minimind#readme", |
|
|
"Bug Tracker": "https://github.com/minimind/minimind/issues", |
|
|
}, |
|
|
packages=find_packages(exclude=["tests", "tests.*", "android", "android.*"]), |
|
|
classifiers=[ |
|
|
"Development Status :: 4 - Beta", |
|
|
"Intended Audience :: Developers", |
|
|
"Intended Audience :: Science/Research", |
|
|
"License :: OSI Approved :: Apache Software License", |
|
|
"Operating System :: OS Independent", |
|
|
"Programming Language :: Python :: 3", |
|
|
"Programming Language :: Python :: 3.9", |
|
|
"Programming Language :: Python :: 3.10", |
|
|
"Programming Language :: Python :: 3.11", |
|
|
"Programming Language :: Python :: 3.12", |
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence", |
|
|
], |
|
|
python_requires=">=3.9", |
|
|
install_requires=[ |
|
|
"torch>=2.1.0", |
|
|
"numpy>=1.24.0", |
|
|
], |
|
|
extras_require={ |
|
|
"train": [ |
|
|
"transformers>=4.35.0", |
|
|
"datasets>=2.14.0", |
|
|
"accelerate>=0.24.0", |
|
|
"wandb>=0.15.0", |
|
|
], |
|
|
"export": [ |
|
|
"onnx>=1.14.0", |
|
|
"onnxruntime>=1.16.0", |
|
|
], |
|
|
"dev": [ |
|
|
"pytest>=7.4.0", |
|
|
"black>=23.0.0", |
|
|
"isort>=5.12.0", |
|
|
"mypy>=1.5.0", |
|
|
], |
|
|
"all": [ |
|
|
"transformers>=4.35.0", |
|
|
"datasets>=2.14.0", |
|
|
"accelerate>=0.24.0", |
|
|
"wandb>=0.15.0", |
|
|
"onnx>=1.14.0", |
|
|
"onnxruntime>=1.16.0", |
|
|
"pytest>=7.4.0", |
|
|
"black>=23.0.0", |
|
|
], |
|
|
}, |
|
|
entry_points={ |
|
|
"console_scripts": [ |
|
|
"minimind-train=scripts.train:main", |
|
|
"minimind-export=scripts.export:main", |
|
|
], |
|
|
}, |
|
|
include_package_data=True, |
|
|
zip_safe=False, |
|
|
) |
|
|
|
