pyproject.toml

[project]
name = "xturing"
version = "0.1.8"
description = "Fine-tuning, evaluation and data generation for LLMs"

authors = [
    { name = "Glenn Ko", email = "glenn@stochastic.ai" },
    { name = "Yuji Chai", email = "yuji.chai@stochastic.ai" },
    { name = "Roman Ageev", email = "roman.ageev@stochastic.ai" },
    { name = "Toan Do", email = "toan.do@stochastic.ai" },
    { name = "Marcos R M", email = "marcos.rm@stochastic.ai" },
    { name = "Sarthak Langde", email = "sarthak.langde@stochastic.ai" },
    { name = "Riccardo Romagnoli", email = "riccardo.romagnoli@stochastic.ai" },
    { name = "Subhash G N", email = "subhash.gn@stochastic.ai" },
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.7",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Scientific/Engineering :: Information Analysis",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Topic :: Text Processing :: Linguistic",
]

license = { file = "LICENSE" }
readme = "README.md"
requires-python = ">=3.7"
keywords = [
    "nlp",
    "llm",
    "finetuning",
    "evaluation",
    "data-generation",
    "training",
    "distributed",
]
dependencies = [
    "torch >= 1.9.0",
    "pytorch-lightning",
    "transformers==4.39.3",
    "datasets==2.14.5",
    "evaluate==0.4.0",
    "bitsandbytes==0.41.1",
    "sentencepiece",
    "deepspeed==0.9.5",
    "gradio",
    "click",
    "wget",
    "ai21",
    "cohere",
    "ipywidgets",
    "openai >= 0.27.0",
    "pydantic >= 1.10.0",
    "rouge-score >= 0.1.2",
    "accelerate==0.22.0",
    "wandb",
]

[project.scripts]
xturing = "xturing.cli:xturing"

[project.optional-dependencies]
int4 = [
    "torch >= 2.0"
]

[project.urls]
homepage = "https://xturing.stochastic.ai/"
documentation = "https://github.com/stochasticai/xturing-docs"
repository = "https://github.com/stochasticai/xturing"


[tool.setuptools.packages.find]
where = ["src"]