|
|
|
|
|
|
|
|
import ast |
|
|
import os |
|
|
import re |
|
|
from pathlib import Path |
|
|
|
|
|
from setuptools import find_packages, setup |
|
|
|
|
|
with open('README.md') as f: |
|
|
long_description = f.read() |
|
|
|
|
|
|
|
|
def get_package_version(): |
|
|
with open(Path(os.path.dirname(os.path.abspath(__file__))) / 'flame' / '__init__.py') as f: |
|
|
version_match = re.search(r"^__version__\s*=\s*(.*)$", f.read(), re.MULTILINE) |
|
|
return ast.literal_eval(version_match.group(1)) |
|
|
|
|
|
|
|
|
setup( |
|
|
name='flame', |
|
|
version=get_package_version(), |
|
|
description='A minimal training framework for scaling FLA models', |
|
|
long_description=long_description, |
|
|
long_description_content_type='text/markdown', |
|
|
author='Songlin Yang, Yu Zhang', |
|
|
author_email='[email protected], [email protected]', |
|
|
url='https://github.com/fla-org/flame', |
|
|
packages=find_packages(), |
|
|
license='MIT', |
|
|
classifiers=[ |
|
|
'Programming Language :: Python :: 3', |
|
|
'License :: OSI Approved :: MIT License', |
|
|
'Operating System :: OS Independent', |
|
|
'Topic :: Scientific/Engineering :: Artificial Intelligence' |
|
|
], |
|
|
python_requires='>=3.10', |
|
|
install_requires=[ |
|
|
'torch>=2.5', |
|
|
'torchdata', |
|
|
'transformers>=4.45.0', |
|
|
'triton>=3.0', |
|
|
'datasets>=3.3.0', |
|
|
'einops', |
|
|
'ninja', |
|
|
'wandb', |
|
|
'tiktoken', |
|
|
'tensorboard', |
|
|
], |
|
|
) |
|
|
|
