Spaces:
Running
on
Zero
Running
on
Zero
| [build-system] | |
| requires = ["setuptools>=61.0"] | |
| build-backend = "setuptools.build_meta" | |
| [project] | |
| name = "cumo" | |
| version = "1.0.0" | |
| description = "CuMo: Scaling Multimodal LLM with Co-Upcycled Mixture-of-Experts." | |
| readme = "README.md" | |
| requires-python = ">=3.8" | |
| classifiers = [ | |
| "Programming Language :: Python :: 3", | |
| "License :: OSI Approved :: Apache Software License", | |
| ] | |
| dependencies = [ | |
| "torch==2.1.2", "torchvision==0.16.2", | |
| "transformers==4.37.2", "tokenizers==0.15.1", "sentencepiece==0.1.99", "shortuuid", | |
| "accelerate==0.21.0", "peft", "bitsandbytes", | |
| "pydantic", "markdown2[all]", "numpy", "scikit-learn==1.2.2", | |
| "gradio==4.16.0", "gradio_client==0.8.1", | |
| "requests", "httpx==0.23.3", "uvicorn", "fastapi", | |
| "einops==0.6.1", "einops-exts==0.0.4", "timm==0.6.13", | |
| ] | |
| [project.optional-dependencies] | |
| train = ["deepspeed", "ninja", "wandb"] | |
| build = ["build", "twine"] | |
| [project.urls] | |
| "Homepage" = "https://github.com/SHI-Labs/CuMo" | |
| "Bug Tracker" = "https://github.com/SHI-Labs/CuMo/issues" | |
| [tool.setuptools.packages.find] | |
| exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"] | |
| [tool.wheel] | |
| exclude = ["assets*", "benchmark*", "docs", "dist*", "playground*", "scripts*", "tests*"] | |