bhadresh-savani
commited on
Commit
•
c668a35
1
Parent(s):
47739cc
added pytorch and tensorflow models
Browse files- config.json +2 -0
- 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,4 +1,5 @@
|
|
1 |
{
|
|
|
2 |
"architectures": [
|
3 |
"T5ForConditionalGeneration"
|
4 |
],
|
@@ -50,6 +51,7 @@
|
|
50 |
"prefix": "translate English to Romanian: "
|
51 |
}
|
52 |
},
|
|
|
53 |
"transformers_version": "4.9.0.dev0",
|
54 |
"use_cache": true,
|
55 |
"vocab_size": 32128
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "./",
|
3 |
"architectures": [
|
4 |
"T5ForConditionalGeneration"
|
5 |
],
|
|
|
51 |
"prefix": "translate English to Romanian: "
|
52 |
}
|
53 |
},
|
54 |
+
"torch_dtype": "float32",
|
55 |
"transformers_version": "4.9.0.dev0",
|
56 |
"use_cache": true,
|
57 |
"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:6ece6e5096ec4abeafde992154a7997ebeb82e7caa1759dec20a0843df8407e3
|
3 |
+
size 891727295
|
tf_model.h5
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:179d50a80941c16d28d2525d9f44e899294c59cdfa9527fe7935c242f641795a
|
3 |
+
size 892144216
|