Spaces:
Running
Running
# torch.ao is a package with a lot of interdependencies. | |
# We will use lazy import to avoid cyclic dependencies here. | |
__all__ = [ | |
"nn", | |
"ns", | |
"quantization", | |
"pruning", | |
] | |
def __getattr__(name): | |
if name in __all__: | |
import importlib | |
return importlib.import_module("." + name, __name__) | |
raise AttributeError(f"module {__name__!r} has no attribute {name!r}") | |