OpenBMB / ModelCenter

Efficient, Low-Resource, Distributed transformer implementation based on BMTrain
https://modelcenter.readthedocs.io
Apache License 2.0
233 stars 28 forks source link

ModelForLM #7

Open QiaoZiqing opened 2 years ago

QiaoZiqing commented 2 years ago

for CPM1 CPM2 Bert GPT2 GPTJ T5 and corresponding return datatype