File size: 494 Bytes
7f1c419 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | 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",
],
},
)
|