Skip to content

GitLab

"test/srt/test_mla_flashinfer.py" did not exist on "33b54e7c40a58d64c988c067e3e607f96a04ae58"
Switch branch/tag
Find file Normal viewHistoryPermalink
resmlp_24.py 168 Bytes
Newer Older
yuguo960516's avatar
bert-large  
yuguo960516 committed
1
2
3
4
5
6
7
8
9
10 
from libai.config import LazyCall
from libai.models import ResMLP

from .resmlp_12 import cfg


cfg.depth = 24
cfg.init_scale = 1e-5

model = LazyCall(ResMLP)(cfg=cfg)