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