File size: 1,179 Bytes
10bcbf6 45cf443 10bcbf6 45cf443 1c5b259 bf14804 45cf443 1aa36e1 45cf443 1aa36e1 45cf443 10bcbf6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 | [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"]
|