birgermoell commited on
Commit
f98be5f
1 Parent(s): 280bd81

Added ppytorch model

Browse files
Files changed (3) hide show
  1. config.json +2 -0
  2. pytorch_model.bin +3 -0
  3. save_torch.py +3 -0
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "architectures": [
3
  "RobertaForMaskedLM"
4
  ],
@@ -18,6 +19,7 @@
18
  "num_hidden_layers": 12,
19
  "pad_token_id": 1,
20
  "position_embedding_type": "absolute",
 
21
  "transformers_version": "4.9.0.dev0",
22
  "type_vocab_size": 1,
23
  "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.9.0.dev0",
24
  "type_vocab_size": 1,
25
  "use_cache": true,
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4ae452e4835ee0e3d42b4f6bb1acd8c272d4ac34fa7e2e2ad79e2bb0a8c94d4
3
+ size 498858859
save_torch.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ from transformers import RobertaForMaskedLM
2
+ model = RobertaForMaskedLM.from_pretrained("./", from_flax=True)
3
+ model.save_pretrained("./")