pytorch model
Browse files- config.json +2 -1
- generate_pytorch_model.py +5 -0
- pytorch_model.bin +3 -0
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"RobertaForMaskedLM"
|
5 |
],
|
@@ -19,6 +19,7 @@
|
|
19 |
"num_hidden_layers": 12,
|
20 |
"pad_token_id": 1,
|
21 |
"position_embedding_type": "absolute",
|
|
|
22 |
"transformers_version": "4.15.0.dev0",
|
23 |
"type_vocab_size": 1,
|
24 |
"use_cache": true,
|
|
|
1 |
{
|
2 |
+
"_name_or_path": ".",
|
3 |
"architectures": [
|
4 |
"RobertaForMaskedLM"
|
5 |
],
|
|
|
19 |
"num_hidden_layers": 12,
|
20 |
"pad_token_id": 1,
|
21 |
"position_embedding_type": "absolute",
|
22 |
+
"torch_dtype": "float32",
|
23 |
"transformers_version": "4.15.0.dev0",
|
24 |
"type_vocab_size": 1,
|
25 |
"use_cache": true,
|
generate_pytorch_model.py
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
# This script overwrites any existing PyTorch model. Generates a new one with an LM head from the pretrained Flax model.
|
2 |
+
from transformers import RobertaForMaskedLM
|
3 |
+
model = RobertaForMaskedLM.from_pretrained(".",from_flax=True)
|
4 |
+
model.save_pretrained(".")
|
5 |
+
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6c7aad3f5456d976b109a957126cbf69ce5c4dcf0ebfd6017269e8ce8a0f9696
|
3 |
+
size 498872555
|