mirror of
https://github.com/linkedin/Liger-Kernel.git
synced 2026-10-12 02:17:47 +00:00
WizyoungandByron Hsu
53dcf02cd2
## Summary Fix `tool.setuptools.packages.find` field in pyproject.toml. Otherwise in local build mode with `pip install .`, python system fails to locate liger_kernel. Co-authored-by: Byron Hsu <[email protected]>
42 lines
863 B
TOML
42 lines
863 B
TOML
[build-system]
|
|
requires = ["setuptools>=42", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "liger_kernel"
|
|
version = "0.2.1"
|
|
description = "Efficient Triton kernels for LLM Training"
|
|
urls = { "Homepage" = "https://github.com/linkedin/Liger-Kernel" }
|
|
readme = { file = "README.md", content-type = "text/markdown" }
|
|
license = { file = "LICENSE" }
|
|
dependencies = [
|
|
"torch>=2.1.2",
|
|
"triton>=2.3.0",
|
|
"transformers>=4.42.0"
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
dev = [
|
|
"matplotlib>=3.7.2",
|
|
"flake8>=4.0.1.1",
|
|
"black>=24.4.2",
|
|
"isort>=5.13.2",
|
|
"pytest>=7.1.2",
|
|
"datasets>=2.19.2",
|
|
"jupyter==1.0.0",
|
|
"seaborn",
|
|
]
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["src"]
|
|
include = ["liger_kernel"]
|
|
|
|
[tool.pytest.ini_options]
|
|
pythonpath = [
|
|
"src",
|
|
"."
|
|
]
|
|
asyncio_mode = "auto"
|
|
log_cli = true
|
|
log_cli_level = "INFO"
|