souging/TRELLIS_TextTo3D

1import importlib2 3__attributes = {4    'SparseStructure': 'sparse_structure',5    6    'SparseFeat2Render': 'sparse_feat2render',7    'SLat2Render':'structured_latent2render',8    'Slat2RenderGeo':'structured_latent2render',9    10    'SparseStructureLatent': 'sparse_structure_latent',11    'TextConditionedSparseStructureLatent': 'sparse_structure_latent',12    'ImageConditionedSparseStructureLatent': 'sparse_structure_latent',13    14    'SLat': 'structured_latent',15    'TextConditionedSLat': 'structured_latent',16    'ImageConditionedSLat': 'structured_latent',17}18 19__submodules = []20 21__all__ = list(__attributes.keys()) + __submodules22 23def __getattr__(name):24    if name not in globals():25        if name in __attributes:26            module_name = __attributes[name]27            module = importlib.import_module(f".{module_name}", __name__)28            globals()[name] = getattr(module, name)29        elif name in __submodules:30            module = importlib.import_module(f".{name}", __name__)31            globals()[name] = module32        else:33            raise AttributeError(f"module {__name__} has no attribute {name}")34    return globals()[name]35 36 37# For Pylance38if __name__ == '__main__':39    from .sparse_structure import SparseStructure40    41    from .sparse_feat2render import SparseFeat2Render42    from .structured_latent2render import (43        SLat2Render,44        Slat2RenderGeo,45    )46    47    from .sparse_structure_latent import (48        SparseStructureLatent,49        TextConditionedSparseStructureLatent,50        ImageConditionedSparseStructureLatent,51    )52    53    from .structured_latent import (54        SLat,55        TextConditionedSLat,56        ImageConditionedSLat,57    )58