diff --git a/lmdeploy/pytorch/engine/engine_checker.py b/lmdeploy/pytorch/engine/engine_checker.py index 1654ece4b5..7276a51fbc 100644 --- a/lmdeploy/pytorch/engine/engine_checker.py +++ b/lmdeploy/pytorch/engine/engine_checker.py @@ -26,7 +26,6 @@ def __init__(self, # pytorch torch_checker = TorchChecker(logger=logger) - self.register_required_checker(torch_checker) if device_type == 'cuda': # triton @@ -39,6 +38,7 @@ def __init__(self, from ..check_env.deeplink import DeeplinkChecker dl_checker = DeeplinkChecker(device_type, logger=logger) self.register_required_checker(dl_checker) + self.register_required_checker(torch_checker) # transformers