/
pyproject.toml
27 lines (23 loc) · 796 Bytes
/
pyproject.toml
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
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "medusa-llm"
version = "1.0"
description = "Simple Framework for Accelerating LLM Generation with Multiple Decoding Heads"
readme = "README.md"
requires-python = ">=3.9"
classifiers = [
"Programming Language :: Python :: 3",
"License :: OSI Approved :: Apache Software License",
]
dependencies = [
"fschat", "torch", "transformers>=4.34", "accelerate", "sentencepiece", "protobuf"
]
[project.optional-dependencies]
train = ["bitsandbytes", "wandb", "scipy"]
[project.urls]
"Homepage" = "https://github.com/FasterDecoding/Medusa"
"Blog" = "https://sites.google.com/view/medusa-llm"
[tool.setuptools.packages.find]
exclude = ["assets*", "notebooks*", "scripts*", "llm_judge"]