🐛 lazy import (#1874)

Mikubill · Aug 5, 2023 · 56c376a · 56c376a
1 parent 82a6213
commit 56c376a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/processor.py b/scripts/processor.py
@@ -5,13 +5,13 @@
 from typing import Callable, Tuple
 
 from modules import safe
-import mmengine
 
 
 def torch_extra_handler(module, name):
     """ Register HistoryBuffer to whitelist, so that A1111 does not complain. 
     This is for mmengine used by DW Pose detector.
     """
+    import mmengine
     if module == 'mmengine.logging.history_buffer' and name in ['HistoryBuffer']:
         return mmengine.logging.history_buffer.HistoryBuffer
     return None