from setuptools import setup, find_packages

setup(
    name="viuai",
    version="1.0.0",
    description="ViuAI-10M: An ultra-deep 32-layer 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",
        "huggingface_hub",
        "datasets",
    ],
    entry_points={
        "console_scripts": [
            "viuai-train=scripts.train:main",
            "viuai-chat=scripts.chat:main",
        ],
    },
)