File size: 1,384 Bytes
0298ad2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
# -*- coding: utf-8 -*-
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',
],
)
|