Jingya
/

tiny-random-bert-remote-code

Feature Extraction

text-embeddings-inference

Inference Endpoints

Model card Files Files and versions Community

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

JingyaHuang

update model

b05d834 over 1 year ago

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")