/ pyproject.toml
pyproject.toml
 1  [build-system]
 2  requires = ["setuptools>=40.6.0", "wheel"]
 3  build-backend = "setuptools.build_meta"
 4  
 5  [project]
 6  name = "janus"
 7  version = "1.0.0"
 8  description = "Janus"
 9  authors = [{name = "DeepSeek-AI"}]
10  license = {file = "LICENSE-CODE"}
11  urls = {homepage = "https://github.com/deepseek-ai/Janus"}
12  readme = "README.md"
13  requires-python = ">=3.8"
14  dependencies = [
15      "torch>=2.0.1",
16      "transformers>=4.38.2",
17      "timm>=0.9.16",
18      "accelerate",
19      "sentencepiece",
20      "attrdict",
21      "einops",
22  ]
23  
24  [project.optional-dependencies]
25  gradio = [
26      "gradio==3.48.0",
27      "gradio-client==0.6.1",
28      "mdtex2html==1.3.0",
29      "pypinyin==0.50.0",
30      "tiktoken==0.5.2",
31      "tqdm==4.64.0",
32      "colorama==0.4.5",
33      "Pygments==2.12.0",
34      "markdown==3.4.1",
35      "SentencePiece==0.1.96"
36  ]
37  lint = [
38      "isort",
39      "black[jupyter] >= 22.6.0",
40      "pylint[spelling] >= 2.15.0",
41      "flake8",
42      "flake8-bugbear",
43      "flake8-comprehensions",
44      "flake8-docstrings",
45      "flake8-pyi",
46      "flake8-simplify",
47      "ruff",
48      "pyenchant",
49      "pre-commit",
50  ]
51  
52  [tool.setuptools]
53  packages = {find = {exclude = ["images"]}}