dk-crazydiv commited on
Commit
aa60c5c
1 Parent(s): 5749ad5

Updated torch model and fixed typo in conversion script

Browse files
Files changed (3) hide show
  1. config.json +2 -0
  2. flax_to_torch.py +1 -1
  3. pytorch_model.bin +2 -2
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,
flax_to_torch.py CHANGED
@@ -11,7 +11,7 @@ def convert_flax_model_to_torch(flax_model_path: str, torch_model_path: str = ".
11
  model.save_pretrained(torch_model_path)
12
 
13
 
14
- if __name__ == "__maim__":
15
  parser = argparse.ArgumentParser(description="Flax to Pytorch model coversion")
16
  parser.add_argument(
17
  "--flax_model_path", type=str, default="flax-community/roberta-pretraining-hindi", help="Flax model path"
 
11
  model.save_pretrained(torch_model_path)
12
 
13
 
14
+ if __name__ == "__main__":
15
  parser = argparse.ArgumentParser(description="Flax to Pytorch model coversion")
16
  parser.add_argument(
17
  "--flax_model_path", type=str, default="flax-community/roberta-pretraining-hindi", help="Flax model path"
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:a73ff85bb66305a29ddd594177637656eee131246ee31d2fae7dfba96cd30755
3
- size 498872747
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d8090645f6bf0629cd8ff4454606d4bd919c1f6f2a09a0eea0bc175050c23f43
3
+ size 498858859