Spaces:
Runtime error
Runtime error
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 | |