Clear redundant codes.

open-mmlab · Sep 1, 2023 · 6135856 · 6135856
1 parent a848134
commit 6135856
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/mmengine/model/base_model/base_model.py b/mmengine/model/base_model/base_model.py
@@ -190,17 +190,15 @@ def to(self, *args, **kwargs) -> nn.Module:
         # directly will cause the NPU to not be found.
         # Here, the input parameters are processed to avoid errors.
         if args and isinstance(args[0], str) and 'npu' in args[0]:
-            import torch_npu
             args = tuple([
                 list(args)[0].replace(
                     'npu', torch.npu.native_device if hasattr(
-                        torch_npu.npu, 'native_device') else 'privateuseone')
+                        torch.npu, 'native_device') else 'privateuseone')
             ])
         if kwargs and 'npu' in str(kwargs.get('device', '')):
-            import torch_npu
             kwargs['device'] = kwargs['device'].replace(
                 'npu', torch.npu.native_device if hasattr(
-                    torch_npu.npu, 'native_device') else 'privateuseone')
+                    torch.npu, 'native_device') else 'privateuseone')
 
         device = torch._C._nn._parse_to(*args, **kwargs)[0]
         if device is not None: