miniLLM/pyproject.toml at master · Micuks/miniLLM · GitHub

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
42
43
44
45
46
47
48
49
50
51
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "miniLLM"
version = "0.3.0"
description = "Text-to-SQL Agent with ReAct reasoning, GRPO RL training, and production serving"
authors = [{ name = "miniLLM" }]
requires-python = ">=3.10"
dependencies = [
    "accelerate>=1.10.1",
    "bitsandbytes>=0.47.0",
    "datasets>=4.0.0",
    "fastapi>=0.116.1",
    "peft>=0.17.1",
    "safetensors>=0.6.2",
    "sentencepiece>=0.2.1",
    "transformers>=4.56.1,<5.0.0",
    "trl>=0.23.0",
    "uvicorn>=0.35.0",
]

[project.optional-dependencies]
quant = [
    "auto-gptq",
    "autoawq",
]
vllm = [
    "vllm",
]
distill = [
    "openai",
]
agent = [
    "openai",
    "deepspeed",
]
bench = [
    "aiohttp",
]
all = [
    "miniLLM[quant,vllm,distill,bench,agent]",
]

[[tool.uv.index]]
url = "https://mirrors.aliyun.com/pypi/simple/"
default = true

[tool.setuptools]
packages = { find = {} }