AbinayaM02
commited on
Commit
•
ed4e9b8
1
Parent(s):
039d504
Converted flax to pytorch model
Browse files- config.json +2 -0
- pytorch_model.bin +3 -0
- src/convert_flax_to_pytorch.py +4 -0
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("../")
|