Views
No views yet
[MASK]. Uncased variants lower-case the input; cased variants preserve case.google-bert/bert-large-cased for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.BertMaskedLM, large cased). Task heads (sequence/token classify, QA, NSP, …) load via hf: fine-tunes.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from zeromodels.models.bert import BertMaskedLM, BertTokenizer
5
6mlm = BertMaskedLM.from_weights("zeromodels/bert_large_cased")
7tokenizer = BertTokenizer.from_weights("zeromodels/bert_large_cased")
8
9inputs = tokenizer("The capital of France is [MASK].")
10logits = mlm(inputs) # (1, L, vocab_size)
11mask = int((inputs["input_ids"][0] == tokenizer.mask_token_id).argmax())
12print(tokenizer.ids_to_tokens[int(logits[0, mask].argmax())])from_weights("zeromodels/<variant>"):| Variant | Hub | Casing |
|---|---|---|
bert_base_uncased | zeromodels/bert_base_uncased | uncased |
bert_large_uncased | zeromodels/bert_large_uncased | uncased |
bert_base_cased | zeromodels/bert_base_cased | cased |
bert_large_cased | zeromodels/bert_large_cased | cased |
from_weights("zeromodels/bert_large_cased") (or on the fly via the hf: prefix). The pretrained backbone is shared; task heads not stored in this checkpoint start randomly initialized, ready for fine-tuning (or load a hf: fine-tune).| Class | Task |
|---|---|
BertModel | Encoder backbone |
BertMaskedLM | Masked language modeling (fill-mask) |
BertSequenceClassify | Sequence classification |
BertTokenClassify | Token classification (NER / POS) |
BertNextSentencePredict | Next-sentence prediction |
BertQnA | Extractive question answering |
BertMultipleChoice | Multiple choice |
1from zeromodels.models.bert import BertSequenceClassify
2model = BertSequenceClassify.from_weights("zeromodels/bert_large_cased")KERAS_BACKEND before importing Keras / zeromodels.BertTokenizer.from_weights(...) so WordPiece casing matches.[MASK] (not <mask>).hf: prefix, e.g. BertMaskedLM.from_weights("hf:google-bert/bert-large-cased").