File size: 430 Bytes
8019be0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
from .mdm import MaskedDiffusionModule
from .any_order import AnyOrderInsertionFlowModule


__all__ = [
    "MaskedDiffusionModule",
    "AutoregressiveModule",
    "AnyOrderInsertionFlowModule",
]


def __getattr__(name):
    if name == "AutoregressiveModule":
        from .autoregressive import AutoregressiveModule
        return AutoregressiveModule
    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")