from libai.config import LazyCall
from libai.models import ResMLP

from .resmlp_12 import cfg


cfg.depth = 36
cfg.init_scale = 1e-6

model = LazyCall(ResMLP)(cfg=cfg)