IMvision12 commited on
Commit
0575ee9
·
verified ·
1 Parent(s): 18b0a76

Add Available classes section to model card

Browse files
Files changed (1) hide show
  1. README.md +18 -0
README.md CHANGED
@@ -68,6 +68,24 @@ Load any DeBERTa variant the same way with `from_weights("kerasformers/<variant>
68
  | `deberta_v3_base` | [`kerasformers/deberta_v3_base`](https://huggingface.co/kerasformers/deberta_v3_base) | v3 |
69
  | `deberta_v3_large` | [`kerasformers/deberta_v3_large`](https://huggingface.co/kerasformers/deberta_v3_large) | v3 |
70
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
71
  ## Tips
72
 
73
  - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
 
68
  | `deberta_v3_base` | [`kerasformers/deberta_v3_base`](https://huggingface.co/kerasformers/deberta_v3_base) | v3 |
69
  | `deberta_v3_large` | [`kerasformers/deberta_v3_large`](https://huggingface.co/kerasformers/deberta_v3_large) | v3 |
70
 
71
+ ## Available classes
72
+
73
+ Load any of these from this repo with `from_weights("kerasformers/deberta_v3_small")` (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).
74
+
75
+ | Class | Task |
76
+ |---|---|
77
+ | `DebertaV3Model` | Encoder backbone |
78
+ | `DebertaV3MaskedLM` | Masked language modeling (fill-mask) |
79
+ | `DebertaV3SequenceClassify` | Sequence classification |
80
+ | `DebertaV3TokenClassify` | Token classification (NER / POS) |
81
+ | `DebertaV3QnA` | Extractive question answering |
82
+ | `DebertaV3MultipleChoice` | Multiple choice |
83
+
84
+ ```python
85
+ from kerasformers.models.deberta_v3 import DebertaV3SequenceClassify
86
+ model = DebertaV3SequenceClassify.from_weights("kerasformers/deberta_v3_small")
87
+ ```
88
+
89
  ## Tips
90
 
91
  - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.