[dlinfer] fix engine checker (#2891)

InternLM · Dec 13, 2024 · 422b9f2 · 422b9f2
1 parent 0749ca5
commit 422b9f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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