Signed-off-by: MengqingCao <cmq0113@163.com>
This commit is contained in:
parent
ea4adeddc1
commit
ccb5376a9a
@ -1,10 +1,12 @@
|
|||||||
import torch
|
import torch
|
||||||
|
|
||||||
import vllm.envs as envs
|
import vllm.envs as envs
|
||||||
from vllm.utils import print_warning_once
|
from vllm.logger import init_logger
|
||||||
|
|
||||||
from .interface import Platform, PlatformEnum
|
from .interface import Platform, PlatformEnum
|
||||||
|
|
||||||
|
logger = init_logger(__name__)
|
||||||
|
|
||||||
|
|
||||||
class OpenVinoPlatform(Platform):
|
class OpenVinoPlatform(Platform):
|
||||||
_enum = PlatformEnum.OPENVINO
|
_enum = PlatformEnum.OPENVINO
|
||||||
@ -27,5 +29,5 @@ class OpenVinoPlatform(Platform):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def is_pin_memory_available(self) -> bool:
|
def is_pin_memory_available(self) -> bool:
|
||||||
print_warning_once("Pin memory is not supported on OpenViNO.")
|
logger.warning("Pin memory is not supported on OpenViNO.")
|
||||||
return False
|
return False
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user