moe-llava
/
pyproject.toml
37 строк · 1.3 Кб
1[build-system]2requires = ["setuptools>=61.0"]3build-backend = "setuptools.build_meta"4
5[project]6name = "moellava"7version = "1.0.0"8description = "Mixture-of-Experts for Large Vision-Language Models."9readme = "README.md"10requires-python = ">=3.8"11classifiers = [12"Programming Language :: Python :: 3",13"License :: OSI Approved :: Apache Software License",14]
15dependencies = [16"torch==2.0.1", "torchvision==0.15.2",17"transformers==4.37.0", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid",18"accelerate==0.21.0", "peft==0.4.0", "bitsandbytes==0.41.0",19"pydantic<2,>=1", "markdown2[all]", "numpy", "scikit-learn==1.2.2",20"gradio==3.37.0", "gradio_client==0.7.0",21"requests", "httpx==0.24.0", "uvicorn", "fastapi",22"einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13",23"protobuf==4.24.4", "tiktoken==0.5.2", "openai==0.28.0", "openpyxl==3.1.2"24]
25
26[project.optional-dependencies]27train = ["deepspeed==0.9.5", "ninja", "wandb", "tensorboardX==2.6.2.2"]28
29[project.urls]30"Homepage" = "https://github.com/PKU-YuanGroup/MoE-LLaVA"31"Bug Tracker" = "https://github.com/PKU-YuanGroup/MoE-LLaVA/issues"32
33[tool.setuptools.packages.find]34exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"]35
36[tool.wheel]37exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"]38