Skip to content

Commit b03e086

Browse files
authored
fix import dlinfer (#16)
1 parent 01a9b4d commit b03e086

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lmdeploy/pytorch/check_env/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ def check_env_deeplink(device_type: str):
3030
if device_type in deeplink_device_type_list:
3131
logger = get_logger('lmdeploy')
3232
try:
33-
import deeplink_ext # noqa: F401
33+
import dlinfer.framework.lmdeploy_ext # noqa: F401
3434
except Exception as e:
3535
_handle_exception(e, 'PyTorch', logger)
3636

0 commit comments

Comments
 (0)