From 41fa84a9a922f19955ebb4265ec19ad10ee89991 Mon Sep 17 00:00:00 2001
From: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Date: Tue, 18 Feb 2025 16:38:51 +0800
Subject: [PATCH] [Fix] remove torch dependencies in `build_function.py`
 (#1632)

---
 mmengine/registry/build_functions.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mmengine/registry/build_functions.py b/mmengine/registry/build_functions.py
index fe907c80..3de67985 100644
--- a/mmengine/registry/build_functions.py
+++ b/mmengine/registry/build_functions.py
@@ -3,8 +3,6 @@ import inspect
 import logging
 from typing import TYPE_CHECKING, Any, Optional, Union
 
-import torch
-
 from mmengine.config import Config, ConfigDict
 from mmengine.utils import ManagerMixin, digit_version
 from .registry import Registry
@@ -238,6 +236,8 @@ def build_optimizer_from_cfg(
         cfg: Union[dict, ConfigDict, Config],
         registry: Registry,
         default_args: Optional[Union[dict, ConfigDict, Config]] = None) -> Any:
+    import torch
+
     from ..logging import print_log
     if 'type' in cfg \
             and 'Adafactor' == cfg['type'] \