NexusCoder / setup.py
AdminReal's picture
Import NexusCoder from github.com/mhieuhonda/NexusCoder
eca5751 verified
Raw
History Blame Contribute Delete
2.24 kB
from setuptools import setup, find_packages
setup(
name="nexus-coder",
version="0.4.0",
description="Nexus Coder v0.4 - CyberForge edition. MoE 423B/39B + 3M context + CyberGym training.",
long_description=open("README.md", "r", encoding="utf-8").read() if __import__("os").path.exists("README.md") else "",
long_description_content_type="text/markdown",
author="Hieu Louis",
author_email="mhieuhonda@users.noreply.github.com",
url="https://github.com/mhieuhonda/NexusCoder",
license="NAL-1.0 (Attribution Required)",
packages=find_packages(),
python_requires="==3.12.13",
install_requires=[
"torch>=2.0.0",
"numpy>=1.24.0",
"tqdm>=4.65.0",
"pyyaml>=6.0",
"datasets>=2.14.0",
"requests>=2.31.0",
"cryptography>=41.0.0",
],
extras_require={
"gpu": ["flash-attn>=2.0.0", "bitsandbytes>=0.41.0", "triton>=2.0.0"],
"data": ["datasets>=2.14.0", "datasketch>=1.6.0", "langdetect>=1.0.9"],
"tools": ["ruff>=0.1.0", "black>=23.0.0", "isort>=5.12.0",
"sqlparse>=0.4.4", "jsbeautifier>=1.14.0"],
"crypto": ["cryptography>=41.0.0", "pyjwt>=2.8.0"],
"database": [
"sqlalchemy>=2.0.0", "psycopg2-binary>=2.9.0", "pymysql>=1.1.0",
"redis>=5.0.0", "pymongo>=4.5.0", "elasticsearch>=8.0.0",
"kafka-python>=2.0.2", "pika>=1.3.0",
],
"web": ["aiohttp>=3.9.0", "websockets>=12.0", "grpcio>=1.59.0",
"beautifulsoup4>=4.12.0", "lxml>=4.9.0"],
"devops": ["paramiko>=3.4.0", "kubernetes>=28.1.0", "docker>=7.0.0"],
"media": ["Pillow>=10.0.0", "reportlab>=4.0.0", "markdown>=3.5.0"],
"ml": ["scikit-learn>=1.3.0", "scipy>=1.11.0", "transformers>=4.35.0",
"accelerate>=0.24.0", "peft>=0.6.0"],
"distributed": ["deepspeed>=0.12.0", "accelerate>=0.24.0", "flash-attn>=2.0.0"],
},
classifiers=[
"Development Status :: 4 - Beta",
"License :: Other/Proprietary License",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.12.13",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
)