Back to all models
Model: julien-c/bert-xsmall-dummy

Monthly model downloads

julien-c/bert-xsmall-dummy julien-c/bert-xsmall-dummy
- downloads
last 30 days

pytorch

tf

Contributed by

julien-c Julien Chaumond company
4 models

How to use this model directly from the 🤗/transformers library:

			
Copy model
tokenizer = AutoTokenizer.from_pretrained("julien-c/bert-xsmall-dummy") model = AutoModelWithLMHead.from_pretrained("julien-c/bert-xsmall-dummy")

How to build a dummy model

from transformers.configuration_bert import BertConfig
from transformers.modeling_bert import BertForMaskedLM
from transformers.modeling_tf_bert import TFBertForMaskedLM
from transformers.tokenization_bert import BertTokenizer


SMALL_MODEL_IDENTIFIER = "julien-c/bert-xsmall-dummy"
DIRNAME = "./bert-xsmall-dummy"

config = BertConfig(10, 20, 1, 1, 40)

model = BertForMaskedLM(config)
model.save_pretrained(DIRNAME)

tf_model = TFBertForMaskedLM.from_pretrained(DIRNAME, from_pt=True)
tf_model.save_pretrained(DIRNAME)

# Slightly different for tokenizer.
# tokenizer = BertTokenizer.from_pretrained(DIRNAME)
# tokenizer.save_pretrained()