bhadresh-savani
commited on
Commit
·
5e5f37d
1
Parent(s):
768c094
added files
Browse files- config.json +2 -1
- convert_to_pytorch.py +3 -0
- convert_to_tensorflow.py +3 -0
- pytorch_model.bin +3 -0
- tf_model.h5 +3 -0
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
@@ -22,6 +22,7 @@
|
|
22 |
"pad_token_id": 0,
|
23 |
"relative_attention_num_buckets": 32,
|
24 |
"tie_word_embeddings": false,
|
|
|
25 |
"transformers_version": "4.9.0.dev0",
|
26 |
"use_cache": true,
|
27 |
"vocab_size": 32128
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "./",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
|
|
22 |
"pad_token_id": 0,
|
23 |
"relative_attention_num_buckets": 32,
|
24 |
"tie_word_embeddings": false,
|
25 |
+
"torch_dtype": "float32",
|
26 |
"transformers_version": "4.9.0.dev0",
|
27 |
"use_cache": true,
|
28 |
"vocab_size": 32128
|
convert_to_pytorch.py
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import AutoModelForSeq2SeqLM
|
2 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("./", from_flax=True)
|
3 |
+
model.save_pretrained("./")
|
convert_to_tensorflow.py
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import TFAutoModelForSeq2SeqLM
|
2 |
+
model = TFAutoModelForSeq2SeqLM.from_pretrained("./", from_pt=True)
|
3 |
+
model.save_pretrained("./")
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:5af83fee0baf991754115deb6cfd949ccb26f6edef3c36eec0b1f61229969bbd
|
3 |
+
size 990434381
|
tf_model.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:41ecd7c3331d83c6432b2279333864907400d9f855713a18b9627b48ac318e0d
|
3 |
+
size 990885200
|