diff --git a/vllm_kunlun/ops/_kunlun_ops.py b/vllm_kunlun/ops/_kunlun_ops.py index 41c6bf2..32a4ffd 100644 --- a/vllm_kunlun/ops/_kunlun_ops.py +++ b/vllm_kunlun/ops/_kunlun_ops.py @@ -25,6 +25,7 @@ from vllm.logger import init_logger logger = init_logger(__name__) try: + import cocopod # noqa import kunlun_ops logger.info("Load custom ops library success!")