File size: 989 Bytes
7c50656 | 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 | [build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "specforge"
dynamic = ["version"]
readme = "README.md"
requires-python = ">=3.11"
description = "SpecForge: Speculative Decoding Training Framework"
authors = [{name = "SGLang Team"}]
urls = {Homepage = "https://github.com/sgl-project/SpecForge"}
dependencies = [
"pre-commit",
"torch==2.9.1",
"torchaudio==2.9.1",
"torchvision==0.24.1",
"transformers==4.57.1",
"qwen-vl-utils==0.0.11",
"datasets",
"setuptools",
"tqdm",
"wandb",
"psutil",
"numpy",
"accelerate",
"pydantic",
"sglang==0.5.9",
"openai-harmony",
"ninja",
"packaging",
"yunchang",
"tensorboard",
]
[tool.setuptools.packages.find]
exclude = ["configs*", "scripts*", "tests*"]
[project.optional-dependencies]
dev = [
"pre-commit",
"unittest"
]
fa = ["flash-attn"]
[tool.setuptools.dynamic]
version = {file = "version.txt"}
|