Views
No views yet
deberta / deberta_v2 / deberta_v3 to match the generation.microsoft/deberta-v2-xxlarge for zeromodels. One implementation runs unmodified on TensorFlow / Torch / JAX.DebertaV2MaskedLM, v2 xxlarge). Task heads (sequence/token classify, QA, …) load via hf: fine-tunes.1import os
2os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
3
4from zeromodels.models.deberta_v2 import (
5 DebertaV2MaskedLM,
6 DebertaV2Tokenizer,
7)
8
9mlm = DebertaV2MaskedLM.from_weights("zeromodels/deberta_v2_xxlarge")
10tokenizer = DebertaV2Tokenizer.from_weights("zeromodels/deberta_v2_xxlarge")
11
12inputs = tokenizer("The capital of France is [MASK].")
13logits = mlm(inputs) # (1, L, vocab_size)
14mask = int((inputs["input_ids"][0] == tokenizer.mask_token_id).argmax())
15print(tokenizer.decode([int(logits[0, mask].argmax())]))from_weights("zeromodels/<variant>"):| Variant | Hub | Generation |
|---|---|---|
deberta_base | zeromodels/deberta_base | v1 |
deberta_large | zeromodels/deberta_large | v1 |
deberta_v2_xlarge | zeromodels/deberta_v2_xlarge | v2 |
deberta_v2_xxlarge | zeromodels/deberta_v2_xxlarge | v2 |
deberta_v3_xsmall | zeromodels/deberta_v3_xsmall | v3 |
deberta_v3_small | zeromodels/deberta_v3_small | v3 |
deberta_v3_base | zeromodels/deberta_v3_base | v3 |
deberta_v3_large | zeromodels/deberta_v3_large | v3 |
from_weights("zeromodels/deberta_v2_xxlarge") (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 |
|---|---|
DebertaV2Model | Encoder backbone |
DebertaV2MaskedLM | Masked language modeling (fill-mask) |
DebertaV2SequenceClassify | Sequence classification |
DebertaV2TokenClassify | Token classification (NER / POS) |
DebertaV2QnA | Extractive question answering |
DebertaV2MultipleChoice | Multiple choice |
1from zeromodels.models.deberta_v2 import DebertaV2SequenceClassify
2model = DebertaV2SequenceClassify.from_weights("zeromodels/deberta_v2_xxlarge")KERAS_BACKEND before importing Keras / zeromodels.Tokenizer.from_weights(...) so vocab and mask token match.hf: prefix, e.g. DebertaV2MaskedLM.from_weights("hf:microsoft/deberta-v2-xxlarge").