[Bugfix][OpenVINO] fix circular reference vllm-project#9939

Signed-off-by: MengqingCao <[email protected]>
MengqingCao · Nov 4, 2024 · 3016837 · 3016837
1 parent 91c9ebb
commit 3016837
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/vllm/platforms/openvino.py b/vllm/platforms/openvino.py
@@ -1,10 +1,12 @@
 import torch
 
 import vllm.envs as envs
-from vllm.utils import print_warning_once
+from vllm.logger import init_logger
 
 from .interface import Platform, PlatformEnum
 
+logger = init_logger(__name__)
+
 
 class OpenVinoPlatform(Platform):
     _enum = PlatformEnum.OPENVINO
@@ -27,5 +29,5 @@ def is_openvino_gpu(self) -> bool:
 
     @classmethod
     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