forked from EngineX-Cambricon/enginex-mlu370-vllm
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"
|