IMvision12 commited on
Commit
8c1ef4e
·
0 Parent(s):

Super-squash branch 'main' using huggingface_hub

Browse files
Files changed (5) hide show
  1. .gitattributes +35 -0
  2. README.md +90 -0
  3. kf_config.json +23 -0
  4. model.weights.h5 +3 -0
  5. tokenizer.json +0 -0
.gitattributes ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,90 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ pipeline_tag: fill-mask
3
+ license: mit
4
+ base_model: FacebookAI/roberta-base
5
+ library_name: kerasformers
6
+ tags:
7
+ - keras
8
+ - kerasformers
9
+ - roberta
10
+ - fill-mask
11
+ - text-encoder
12
+ - arxiv:1907.11692
13
+ - pytorch
14
+ - jax
15
+ - tf
16
+ ---
17
+
18
+ ## ***See [our collection](https://huggingface.co/collections/kerasformers/roberta-6a6e8d9b2f4c9253f4145f65) for all versions of RoBERTa.***
19
+
20
+ # Run RoBERTa with Keras 3: JAX, PyTorch, or TensorFlow
21
+
22
+ [![GitHub](https://img.shields.io/badge/GitHub-KerasFormers-black?logo=github)](https://github.com/IMvision12/KerasFormers) [![Docs](https://img.shields.io/badge/Docs-RoBERTa-blue)](https://imvision12.github.io/KerasFormers/roberta/) [![Collection](https://img.shields.io/badge/HF-RoBERTa%20collection-yellow)](https://huggingface.co/collections/kerasformers/roberta-6a6e8d9b2f4c9253f4145f65)
23
+
24
+ # kerasformers/roberta_base
25
+
26
+ Paper: [RoBERTa: A Robustly Optimized BERT Pretraining Approach (arXiv:1907.11692)](https://arxiv.org/abs/1907.11692) · [HF Papers](https://huggingface.co/papers/1907.11692)
27
+
28
+ RoBERTa is a robustly optimized BERT encoder: more data/steps, no NSP, dynamic masking, byte-level BPE (mask token `<mask>`), and padding-offset position ids.
29
+
30
+ For more details on the model, please go to the upstream [model card](https://huggingface.co/FacebookAI/roberta-base).
31
+
32
+ Pure-**Keras 3** conversion of [`FacebookAI/roberta-base`](https://huggingface.co/FacebookAI/roberta-base) for [kerasformers](https://github.com/IMvision12/KerasFormers). One implementation runs unmodified on **TensorFlow / Torch / JAX**.
33
+
34
+ This is a **fill-mask / encoder** checkpoint (`RobertaMaskedLM`, base). Task heads load via `hf:` fine-tunes.
35
+
36
+ ## ✨ Quick start (fill-mask)
37
+
38
+ ```python
39
+ import os
40
+ os.environ["KERAS_BACKEND"] = "torch" # or "jax" / "tensorflow"
41
+
42
+ from kerasformers.models.roberta import RobertaMaskedLM, RobertaTokenizer
43
+
44
+ mlm = RobertaMaskedLM.from_weights("kerasformers/roberta_base")
45
+ tokenizer = RobertaTokenizer.from_weights("kerasformers/roberta_base")
46
+
47
+ inputs = tokenizer("The capital of France is <mask>.")
48
+ logits = mlm(inputs) # (1, L, vocab_size)
49
+ mask = int((inputs["input_ids"][0] == tokenizer.mask_token_id).argmax())
50
+ print(tokenizer.decode([int(logits[0, mask].argmax())]))
51
+ ```
52
+
53
+ Load any RoBERTa variant the same way with `from_weights("kerasformers/<variant>")`:
54
+
55
+ | Variant | Hub |
56
+ |---|---|
57
+ | `roberta_base` | [`kerasformers/roberta_base`](https://huggingface.co/kerasformers/roberta_base) |
58
+ | `roberta_large` | [`kerasformers/roberta_large`](https://huggingface.co/kerasformers/roberta_large) |
59
+
60
+ ## Available classes
61
+
62
+ Load any of these from this repo with `from_weights("kerasformers/roberta_base")` (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).
63
+
64
+ | Class | Task |
65
+ |---|---|
66
+ | `RobertaModel` | Encoder backbone |
67
+ | `RobertaMaskedLM` | Masked language modeling (fill-mask) |
68
+ | `RobertaSequenceClassify` | Sequence classification |
69
+ | `RobertaTokenClassify` | Token classification (NER / POS) |
70
+ | `RobertaQnA` | Extractive question answering |
71
+ | `RobertaMultipleChoice` | Multiple choice |
72
+
73
+ ```python
74
+ from kerasformers.models.roberta import RobertaSequenceClassify
75
+ model = RobertaSequenceClassify.from_weights("kerasformers/roberta_base")
76
+ ```
77
+
78
+ ## Tips
79
+
80
+ - Set `KERAS_BACKEND` **before** importing Keras / kerasformers.
81
+ - Prefer `RobertaTokenizer.from_weights(...)` so BPE vocab matches.
82
+ - Use `<mask>` (not `[MASK]`).
83
+ - See [RoBERTa docs](https://imvision12.github.io/KerasFormers/roberta/) and [Loading Weights](https://imvision12.github.io/KerasFormers/loading_weights/).
84
+ - Community / upstream safetensors still work via the `hf:` prefix, e.g. `RobertaMaskedLM.from_weights("hf:FacebookAI/roberta-base")`.
85
+
86
+ ## Special Thanks
87
+
88
+ A huge thank you to the Facebook AI RoBERTa authors for creating and releasing these models.
89
+
90
+ License: MIT.
kf_config.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "library_name": "kerasformers",
3
+ "kerasformers_version": "1.2.1",
4
+ "model_module": "kerasformers.models.roberta",
5
+ "model_class": "RobertaModel",
6
+ "variant": "roberta_base",
7
+ "weights": "model.weights.h5",
8
+ "schema_version": 2,
9
+ "weight_dtype": "float32",
10
+ "model_type": "roberta",
11
+ "text_config": {
12
+ "vocab_size": 50265,
13
+ "embed_dim": 768,
14
+ "num_layers": 12,
15
+ "num_heads": 12,
16
+ "mlp_dim": 3072,
17
+ "max_position_embeddings": 514,
18
+ "type_vocab_size": 1,
19
+ "hidden_act": "gelu",
20
+ "layer_norm_eps": 1e-05,
21
+ "pad_token_id": 1
22
+ }
23
+ }
model.weights.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:446246ae06a71fb171530c0a15941d6867dded4028917975a61d62d009ab8e2c
3
+ size 656034460
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff