Medusa/pyproject.toml at main · FasterDecoding/Medusa · GitHub

27 lines (23 loc) · 796 Bytes

[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"]

Expand file tree