142 lines
3.4 KiB
TOML
142 lines
3.4 KiB
TOML
[project]
|
|
name = "openmind"
|
|
dynamic = ["version"]
|
|
description = "openMind is a magicain who takes you to experience the mystery and creativity of AI."
|
|
readme = "README.md"
|
|
license = { text = "木兰宽松许可证， 第2版" }
|
|
authors = [
|
|
{ name = "The openMind Team" },
|
|
{ email = "dev@public.openmind.cn" },
|
|
]
|
|
dependencies = ["tqdm",
|
|
"sentencepiece",
|
|
"tokenizers",
|
|
"tabulate",
|
|
"decorator",
|
|
"tiktoken",
|
|
"packaging",
|
|
"docker >= 4.34.2",
|
|
"einops == 0.8.0",
|
|
"setuptools == 69.5.1",
|
|
"datasets >= 2.18.0, <= 2.21.0",
|
|
"pyarrow == 16.1.0",
|
|
"openmind-hub >= 0.9.1",
|
|
"numpy < 2.0.0"]
|
|
requires-python = ">= 3.8, < 3.12"
|
|
classifiers = [
|
|
"Development Status :: 1 - Planning",
|
|
"Intended Audience :: Developers",
|
|
"Intended Audience :: Education",
|
|
"Intended Audience :: Science/Research",
|
|
"Operating System :: OS Independent",
|
|
"Programming Language :: Python :: 3",
|
|
"Programming Language :: Python :: 3.8",
|
|
"Programming Language :: Python :: 3.9",
|
|
"Programming Language :: Python :: 3.10",
|
|
"Programming Language :: Python :: 3.11",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
"License :: OSI Approved :: Mulan Permissive Software License v2 (MulanPSL-2.0)",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
test = [
|
|
"pytest",
|
|
"pytest-cov",
|
|
"testtools",
|
|
"openmind[pt-cpu]",
|
|
"pillow",
|
|
"lm_eval == 0.4.3",
|
|
]
|
|
|
|
lint = [
|
|
"black == 24.8.0",
|
|
"ruff == 0.6.1",
|
|
]
|
|
|
|
ci = [
|
|
"requests"
|
|
]
|
|
|
|
pt-cpu = [
|
|
"torch == 2.1.0",
|
|
"transformers >= 4.51.1, <=4.51.3",
|
|
"accelerate >= 1.0.0rc1, <= 1.7.0",
|
|
"deepspeed >= 0.15.4, <= 0.16.9",
|
|
"evaluate == 0.4.1",
|
|
"lm_eval == 0.4.3",
|
|
"diffusers >= 0.29.0, <= 0.31.0",
|
|
"peft >= 0.12.0",
|
|
"trl == 0.9.3",
|
|
]
|
|
|
|
pt = [
|
|
"openmind[pt-cpu]",
|
|
"torch-npu == 2.1.0.post10",
|
|
]
|
|
|
|
ms = [
|
|
"mindspore == 2.4.10",
|
|
"mindformers == 1.3.2",
|
|
]
|
|
|
|
[project.urls]
|
|
Homepage = "https://gitee.com/openmind-ai/openmind"
|
|
Repository = "https://gitee.com/openmind-ai/openmind"
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[tool.hatch.metadata]
|
|
allow-direct-references = true
|
|
|
|
[tool.hatch.build]
|
|
include = [
|
|
"src/*","examples/*","sources/*"
|
|
]
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
packages = ["src/openmind"]
|
|
|
|
# Used to call hatch_build.py
|
|
[tool.hatch.build.targets.wheel.hooks.custom]
|
|
|
|
# Used to get version information
|
|
[tool.hatch.metadata.hooks.custom]
|
|
#path = "custom_metadata_hook.py"
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ['py38']
|
|
include = '\.pyi?$'
|
|
|
|
[tool.ruff]
|
|
line-length = 120
|
|
# Enables the isort rules on top of the default rules
|
|
|
|
# "F401": Ignore import vilations in all `__init__.py` files.
|
|
# "I": Ignore isort in all `__init__.py` files to avoid circle import.
|
|
[tool.ruff.lint.per-file-ignores]
|
|
"__init__.py" = ["F401", "I"]
|
|
|
|
[tool.ruff.lint.isort]
|
|
lines-after-imports = 2
|
|
# Sort the imports by module, independent for import style (from / import).
|
|
force-sort-within-sections = true
|
|
|
|
[tool.ruff.format]
|
|
# Like Black, use double quotes for strings.
|
|
quote-style = "double"
|
|
|
|
# Like Black, indent with spaces, rather than tabs.
|
|
indent-style = "space"
|
|
|
|
# Like Black, respect magic trailing commas.
|
|
skip-magic-trailing-comma = false
|
|
|
|
# Like Black, automatically detect the appropriate line ending.
|
|
line-ending = "auto"
|
|
|
|
[project.scripts]
|
|
openmind-cli = "openmind.cli.cli:main"
|