pere commited on
Commit
3c17867
1 Parent(s): 219a065
Files changed (3) hide show
  1. config.json +1 -1
  2. convert_model.py +4 -0
  3. pytorch_model.bin +2 -2
config.json CHANGED
@@ -1,7 +1,7 @@
1
  {
2
  "_name_or_path": ".",
3
  "architectures": [
4
- "XLMRobertaModel"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 0,
 
1
  {
2
  "_name_or_path": ".",
3
  "architectures": [
4
+ "XLMRobertaForMaskedLM"
5
  ],
6
  "attention_probs_dropout_prob": 0.1,
7
  "bos_token_id": 0,
convert_model.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from transformers import XLMRobertaForMaskedLM
2
+ model = XLMRobertaForMaskedLM.from_pretrained(".")
3
+ model.save_pretrained(".")
4
+
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:2750c92451f167d9a2b7d65654918db9bf946324f415190bbd8d55c4c33b1ce8
3
- size 1112242989
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:221be2249c32d705bc1bf33ee7c96b78bb8750effd1e512195c3e494bcca8e0b
3
+ size 1113251641