| from setuptools import setup, find_packages | |
| setup( | |
| name="viuai", | |
| version="0.1.0", | |
| description="ViuAI-1M: A lightweight chat language model built from scratch", | |
| author="ViuAI Team", | |
| packages=find_packages(), | |
| python_requires=">=3.9", | |
| install_requires=[ | |
| "torch>=2.0.0", | |
| "numpy>=1.24.0", | |
| ], | |
| entry_points={ | |
| "console_scripts": [ | |
| "viuai-train=scripts.train:main", | |
| "viuai-chat=scripts.chat:main", | |
| ], | |
| }, | |
| ) | |