Jingya
/

tiny-random-bert-remote-code

Feature Extraction

text-embeddings-inference

Model card Files Files and versions

tiny-random-bert-remote-code / create_model.py

fuzeey's picture

update model

b05d834 about 3 years ago

History Blame Contribute Delete

387 Bytes

	# from transformers import AutoConfig

	# from modeling.modeling_bert import BertCustomLMHeadModel

	# cfg = AutoConfig.from_pretrained("hf-internal-testing/tiny-random-BertModel")

	# BertCustomLMHeadModel.register_for_auto_class("AutoModelForSequenceClassification")

	# model = BertCustomLMHeadModel(cfg)
	# model.save_pretrained("/home/Jingya/hf_internship/tiny-testing-gpt2-remote-code")