File size: 980 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 |
[project]
name = "flame"
dynamic = ["version"]
description = "A minimal training framework for scaling FLA models"
readme = "README.md"
authors = [
{ name = "Songlin Yang", email = "[email protected]" },
{ name = "Yu Zhang", email = "[email protected]" },
]
license = { file = "LICENSE" }
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: MIT License",
"Operating System :: OS Independent",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
]
requires-python = ">=3.10"
dependencies = [
'torch>=2.5',
'torchdata',
'transformers>=4.45.0',
'triton>=3.0',
'datasets>=3.3.0',
'einops',
'ninja',
'wandb',
'tiktoken',
'tensorboard',
]
[project.optional-dependencies]
dev = ["pytest"]
[project.urls]
Homepage = "https://github.com/fla-org/flame"
[build-system]
requires = ["setuptools>=45", "wheel", "ninja", "torch"]
[tool.isort]
line_length = 127
multi_line_output = 3
|