instant-nsr-pl/models/__init__.py

models = {}


def register(name):
    def decorator(cls):
        models[name] = cls
        return cls
    return decorator


def make(name, config):
    model = models[name](config)
    return model


from . import nerf, neus, geometry, texture