/
githubmirror
/
ColossalAI
Обзор
Документация
Войти
/
githubmirror
/
ColossalAI
Код
Запросы
0
Пакеты
0
Релизы
0
Аналитика
Безопасность
main
applications/ColossalMoE/setup.py
43 строки
1 KB
Xuanlei Zhao
[moe] init mixtral impl
07 фев 2024, 14:21
07 фев 2024, 14:21
7d8e033
Код
Авторство
О чём код?
from setuptools import find_packages, setup def fetch_requirements(path): with open(path, "r") as fd: return [r.strip() for r in fd.readlines()] def fetch_readme(): with open("README.md", encoding="utf-8") as f: return f.read() def fetch_version(): with open("version.txt", "r") as f: return f.read().strip() setup( name="colossal_moe", version=fetch_version(), packages=find_packages( exclude=( "tests", "benchmarks", "*.egg-info", ) ), description="Colossal-AI MoE", long_description=fetch_readme(), long_description_content_type="text/markdown", license="Apache Software License 2.0", url="https://github.com/hpcaitech", install_requires=fetch_requirements("requirements.txt"), python_requires=">=3.6", classifiers=[ "Programming Language :: Python :: 3", "License :: OSI Approved :: Apache Software License", "Environment :: GPU :: NVIDIA CUDA", "Topic :: Scientific/Engineering :: Artificial Intelligence", "Topic :: System :: Distributed Computing", ], )