diff --git a/torch/_dynamo/trace_rules.py b/torch/_dynamo/trace_rules.py index 0698b5fd8f12e..daeb8626c1094 100644 --- a/torch/_dynamo/trace_rules.py +++ b/torch/_dynamo/trace_rules.py @@ -3036,10 +3036,6 @@ def add_module_init_func(name: str, init_func: Callable[[], None]) -> None: """Register a module without eagerly importing it""" # If the module is already imported, eagerly run init assert "." not in name, f"Expected a root module name, but got {name}" - if name in sys.modules: - init_func() - - # Module is not yet imported, delay processing until needed assert name not in _lazy_module_init _lazy_module_init[name].append(init_func)