| |
|
|
| 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='yangsl66@mit.edu, yzhang.cs@outlook.com', |
| 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.6', |
| 'torchdata', |
| 'transformers>=4.45.0', |
| 'triton>=3.0', |
| 'datasets>=3.3.0', |
| 'einops', |
| 'ninja', |
| 'wandb', |
| 'tiktoken', |
| 'tensorboard', |
| ], |
| ) |
|
|
