bart-plugin/setup.py at master · vllm-project/bart-plugin · 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
"""Setup script for vLLM BART model plugin."""

from setuptools import find_packages, setup

setup(
    name="vllm-bart-plugin",
    version="0.4.0",
    description="BART model plugin for vLLM",
    author="Nicolò Lucchesi",
    author_email="nick.lucche@redhat.com",
    packages=find_packages(),
    python_requires=">=3.10",
    install_requires=[
        "vllm>=0.13.0,<=0.19.1",
        "torch>=2.9.0",
        "transformers >= 4.56.0, < 5",
    ],
    entry_points={
        "vllm.general_plugins": [
            "bart=vllm_bart_plugin:register_bart_model",
        ],
    },
    classifiers=[
        "Development Status :: 3 - Alpha",
        "Intended Audience :: Developers",
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.8",
        "Programming Language :: Python :: 3.9",
        "Programming Language :: Python :: 3.10",
        "Programming Language :: Python :: 3.11",
    ],
)