from model.registry import MODEL import model.residual_generator