Janus/pyproject.toml
wuchengyue 659982f413 commit
2024-10-18 11:58:52 +08:00

54 lines
1.1 KiB
TOML

[build-system]
requires = ["setuptools>=40.6.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "janus"
version = "1.0.0"
description = "Janus"
authors = [{name = "DeepSeek-AI"}]
license = {file = "LICENSE-CODE"}
urls = {homepage = "https://github.com/deepseek-ai/Janus"}
readme = "README.md"
requires-python = ">=3.8"
dependencies = [
"torch>=2.0.1",
"transformers>=4.38.2",
"timm>=0.9.16",
"accelerate",
"sentencepiece",
"attrdict",
"einops",
]
[project.optional-dependencies]
gradio = [
"gradio==3.48.0",
"gradio-client==0.6.1",
"mdtex2html==1.3.0",
"pypinyin==0.50.0",
"tiktoken==0.5.2",
"tqdm==4.64.0",
"colorama==0.4.5",
"Pygments==2.12.0",
"markdown==3.4.1",
"SentencePiece==0.1.96"
]
lint = [
"isort",
"black[jupyter] >= 22.6.0",
"pylint[spelling] >= 2.15.0",
"flake8",
"flake8-bugbear",
"flake8-comprehensions",
"flake8-docstrings",
"flake8-pyi",
"flake8-simplify",
"ruff",
"pyenchant",
"pre-commit",
]
[tool.setuptools]
packages = {find = {exclude = ["images"]}}