mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-04-16 22:07:11 +08:00
10 lines
211 B
Python
10 lines
211 B
Python
from .interface import Platform, PlatformEnum
|
|
|
|
|
|
class NeuronPlatform(Platform):
|
|
_enum = PlatformEnum.NEURON
|
|
|
|
@classmethod
|
|
def get_device_name(cls, device_id: int = 0) -> str:
|
|
return "neuron"
|