BMTrain icon indicating copy to clipboard operation
BMTrain copied to clipboard
verified publisher icon OpenBMB

Can BMTrain be directly used for huggingface transformer models?

Open Kunlun-Zhu opened this issue 3 years ago • 1 comments

Kunlun-Zhu avatar Sep 05 '22 08:09 Kunlun-Zhu

Yes, there is a wrapper bmtrain.BMTrainModelWrapper that does this. Usage: bmtrain_model = bmtrain.BMTrainModelWrapper(huggingface_model)

a710128 avatar Sep 08 '22 08:09 a710128