[project]
name = "minimind"
version = "3.0.0"
description = "MiniMind: 从0训练极小参数量的中文大语言模型"
readme = "README.md"
requires-python = ">=3.10,<3.13"
# 与 requirements.txt 保持一致；torch 默认走 PyPI（Linux 为 CUDA 版，macOS 为 MPS 版），
# 如需指定 CUDA 版本可改用 https://download.pytorch.org/whl/ 对应索引
dependencies = [
    "datasets==3.6.0",
    "datasketch==1.6.4",
    "einops==0.8.1",
    "fastapi==0.120.1",
    "flask==3.0.3",
    "flask-cors==4.0.0",
    "jieba==0.42.1",
    "jinja2==3.1.2",
    "jsonlines==4.0.0",
    "marshmallow==3.22.0",
    "modelscope==1.37.0",
    "ngrok==1.4.0",
    "nltk==3.8",
    "numpy==1.26.4",
    "openai==1.59.6",
    "protobuf==6.33.5",
    "psutil==5.9.8",
    "pydantic==2.11.5",
    "rich==13.7.1",
    "scikit-learn==1.5.1",
    "sentence-transformers==2.3.1",
    "sentencepiece==0.2.0",
    "simhash==2.1.2",
    "streamlit==1.50.0",
    "swanlab==0.9.8",
    "tiktoken==0.10.0",
    "torch==2.6.0",
    "transformers==4.57.6",
    "trl==0.13.0",
    "uvicorn==0.38.0",
    "wandb==0.22.3",
]

[tool.uv]
package = false