Jingya/tiny-random-bert-remote-code

1# from transformers import AutoConfig2 3# from modeling.modeling_bert import BertCustomLMHeadModel4 5# cfg = AutoConfig.from_pretrained("hf-internal-testing/tiny-random-BertModel")6 7# BertCustomLMHeadModel.register_for_auto_class("AutoModelForSequenceClassification")8 9# model = BertCustomLMHeadModel(cfg)10# model.save_pretrained("/home/Jingya/hf_internship/tiny-testing-gpt2-remote-code")