AbinayaM02 commited on
Commit
ed4e9b8
1 Parent(s): 039d504

Converted flax to pytorch model

Browse files
config.json CHANGED
@@ -1,4 +1,5 @@
1
  {
 
2
  "activation_function": "gelu_new",
3
  "architectures": [
4
  "GPT2LMHeadModel"
@@ -30,6 +31,7 @@
30
  "max_length": 50
31
  }
32
  },
 
33
  "transformers_version": "4.9.0.dev0",
34
  "use_cache": true,
35
  "vocab_size": 50257
 
1
  {
2
+ "_name_or_path": "../gpt-2-tamil",
3
  "activation_function": "gelu_new",
4
  "architectures": [
5
  "GPT2LMHeadModel"
 
31
  "max_length": 50
32
  }
33
  },
34
+ "torch_dtype": "float32",
35
  "transformers_version": "4.9.0.dev0",
36
  "use_cache": true,
37
  "vocab_size": 50257
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5be21c80df001c71d4550e2c2c5f6e598ca70aeefb6e311ecb68fccc4c151014
3
+ size 510401385
src/convert_flax_to_pytorch.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ from transformers import GPT2LMHeadModel
2
+
3
+ model = GPT2LMHeadModel.from_pretrained("../gpt-2-tamil", from_flax=True)
4
+ model.save_pretrained("../")