Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Otherwise, the [Trainer] won't save the weights in fp16 and it won't create a pytorch_model.bin file.