pyproject.toml

[build-system]
requires = ["setuptools", "setuptools-scm"]
build-backend = "setuptools.build_meta"

[project]
name = "lai-textpred"
authors = [
    {name = "Lightning-AI", email = "name@lightning.ai"},
]
description = "LAI components for training text prediction"
readme = "README.md"
requires-python = ">=3.8"
keywords = ["deep learning", "machine learning", "pytorch", "AI"]
license = {text = "Apache-2.0"}
classifiers = [
    "Programming Language :: Python :: 3",
]
dependencies = [
    "numpy",
    "lightning[app] >=2.1.0",
    "deepspeed ==0.14.0",
    "pynvml",
    "lightning-gpt ==0.1.1",
    "tensorboardX",
]
dynamic = ["version"]

[tool.setuptools.dynamic]
version = {attr = "lai_textpred.__version__"}

[tool.setuptools.packages.find]
include = ["lai_textpred*"]
exclude = ["tests*"]

[project.urls]
repository = "https://github.com/Lightning-Universe/Text-Prediction_component"