[Bugfix] cocopod ops can't be finded (#242)
Signed-off-by: Li Wei <liwei.109@outlook.com>
This commit is contained in:
@@ -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!")
|
||||
|
||||
Reference in New Issue
Block a user