mirror of
https://github.com/handsomezhuzhu/vllm-npu-plugin.git
synced 2026-02-20 11:42:30 +00:00
29 lines
753 B
Python
29 lines
753 B
Python
"""
|
|
vllm-npu-plugin: Ascend NPU platform plugin for vLLM v0.11.0.
|
|
|
|
Registers as an out-of-tree platform plugin via the
|
|
``vllm.platform_plugins`` entry-point group.
|
|
"""
|
|
|
|
from setuptools import find_packages, setup
|
|
|
|
setup(
|
|
name="vllm-npu-plugin",
|
|
version="0.1.0",
|
|
description="Ascend NPU platform plugin for vLLM v0.11.0",
|
|
packages=find_packages(),
|
|
python_requires=">=3.9",
|
|
install_requires=[
|
|
# vllm must already be installed (v0.11.0)
|
|
],
|
|
entry_points={
|
|
"vllm.platform_plugins": [
|
|
"npu = vllm_npu:register",
|
|
],
|
|
"vllm.general_plugins": [
|
|
"npu_enhanced_model = vllm_npu:register_model",
|
|
"npu_kv_connector = vllm_npu:register_connector",
|
|
],
|
|
},
|
|
)
|