63 lines
1.5 KiB
TOML
63 lines
1.5 KiB
TOML
[project]
|
|
name = "minimind-v"
|
|
version = "0.1.0"
|
|
description = "MiniMind-V multimodal vision-language model training and inference scripts."
|
|
readme = "README.md"
|
|
requires-python = ">=3.10,<3.11"
|
|
dependencies = [
|
|
"datasets==3.6.0",
|
|
"datasketch==1.6.4",
|
|
"Flask==3.0.3",
|
|
"Flask-Cors==4.0.0",
|
|
"jieba==0.42.1",
|
|
"jsonlines==4.0.0",
|
|
"marshmallow==3.22.0",
|
|
"matplotlib==3.5.1",
|
|
"ngrok==1.4.0",
|
|
"nltk==3.8",
|
|
"numpy==1.26.4",
|
|
"openai==1.59.6",
|
|
"peft==0.7.1",
|
|
"psutil==5.9.8",
|
|
"pydantic==2.8.2",
|
|
"rich==13.7.1",
|
|
"scikit-learn==1.5.1",
|
|
"sentence-transformers==2.3.1",
|
|
"simhash==2.1.2",
|
|
"tiktoken==0.10.0",
|
|
"transformers==4.57.6",
|
|
"jinja2==3.1.2",
|
|
"trl==0.13.0",
|
|
"ujson==5.1.0",
|
|
"wandb==0.18.3",
|
|
"streamlit==1.50.0",
|
|
"gradio==5.49.1",
|
|
"modelscope-studio==1.6.1",
|
|
"python-multipart==0.0.22",
|
|
"pyarrow==23.0.0",
|
|
"Pillow",
|
|
"swanlab==0.6.12",
|
|
"torch==2.10.0",
|
|
"torchvision==0.25.0",
|
|
]
|
|
|
|
[tool.uv]
|
|
package = false
|
|
|
|
[tool.uv.sources]
|
|
torch = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'win32' or sys_platform == 'linux'" },
|
|
]
|
|
torchvision = [
|
|
{ index = "pytorch-cu128", marker = "sys_platform == 'win32' or sys_platform == 'linux'" },
|
|
]
|
|
|
|
[[tool.uv.index]]
|
|
url = "https://pypi.tuna.tsinghua.edu.cn/simple"
|
|
default = true
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu128"
|
|
url = "https://download.pytorch.org/whl/cu128"
|
|
explicit = true
|