File size: 720 Bytes
13c5606 | 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 | [project]
name = "dflash"
version = "0.1.0"
description = "DFlash: Block Diffusion for Flash Speculative Decoding"
requires-python = ">=3.10"
dependencies = [
"rich",
"loguru",
"numpy",
"tqdm",
"datasets",
"requests",
"huggingface-hub",
]
[tool.setuptools.packages.find]
include = ["dflash*"]
[project.optional-dependencies]
transformers = [
"torch",
"transformers==4.57.1",
"accelerate",
"typing-extensions",
]
sglang = [
"sglang[all] @ git+https://github.com/sgl-project/sglang.git@refs/pull/23000/head#subdirectory=python",
"transformers",
]
vllm = [
"vllm",
"datasets>=3,<4",
"huggingface-hub<1",
]
mlx = [
"mlx==0.31.2",
"mlx-lm==0.31.3",
]
|