| [project] | |
| name = "omni" | |
| version = "0.1.0" | |
| description = "MiniMind LLM (and future multimodal) training & inference framework, structured as an installable package." | |
| readme = "README.md" | |
| requires-python = ">=3.12" | |
| dependencies = [ | |
| "torch>=2.6.0", | |
| "transformers>=4.57.6", | |
| "datasets>=3.6.0", | |
| "tokenizers>=0.20.0", | |
| "tiktoken>=0.10.0", | |
| "jinja2>=3.1.2", | |
| "numpy>=1.26.4", | |
| "safetensors>=0.5.0", | |
| "accelerate>=1.0.0", | |
| "sentencepiece>=0.2.0", | |
| "jieba>=0.42.1", | |
| "pydantic>=2.11.5", | |
| "rich>=13.7.1", | |
| "regex", | |
| "pyyaml>=6.0.2", | |
| "pillow>=10.4.0", | |
| "pyarrow>=17.0.0", | |
| "requests>=2.32.3", | |
| "pydub>=0.25.1", | |
| ] | |
| [project.optional-dependencies] | |
| rl = ["swanlab>=0.7.11", "wandb>=0.18.3"] | |
| serve = ["fastapi>=0.115.0", "uvicorn", "pydantic>=2.11.5", "openai>=1.59.6", "flask>=3.0.3", "flask-cors>=4.0.0"] | |
| demo = ["streamlit>=1.50.0"] | |
| all = ["swanlab>=0.7.11", "wandb>=0.18.3", "fastapi>=0.115.0", "uvicorn", "openai>=1.59.6", "flask>=3.0.3", "flask-cors>=4.0.0", "streamlit>=1.50.0", "sglang"] | |
| [build-system] | |
| requires = ["setuptools>=61.0"] | |
| build-backend = "setuptools.build_meta" | |
| [tool.setuptools.packages.find] | |
| where = ["src"] | |