PyTorch models converted
Browse files- config.json +2 -2
- convert_to_pytorch.py +5 -0
- generation_config.json +1 -1
- pytorch_model.bin +3 -0
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "gelu",
|
5 |
"apply_spec_augment": false,
|
@@ -146,7 +146,7 @@
|
|
146 |
50362
|
147 |
],
|
148 |
"torch_dtype": "float32",
|
149 |
-
"transformers_version": "4.
|
150 |
"use_cache": true,
|
151 |
"use_weighted_layer_sum": false,
|
152 |
"vocab_size": 51865
|
|
|
1 |
{
|
2 |
+
"_name_or_path": ".",
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "gelu",
|
5 |
"apply_spec_augment": false,
|
|
|
146 |
50362
|
147 |
],
|
148 |
"torch_dtype": "float32",
|
149 |
+
"transformers_version": "4.30.1",
|
150 |
"use_cache": true,
|
151 |
"use_weighted_layer_sum": false,
|
152 |
"vocab_size": 51865
|
convert_to_pytorch.py
ADDED
@@ -0,0 +1,5 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
from transformers import WhisperForConditionalGeneration
|
2 |
+
model = WhisperForConditionalGeneration.from_pretrained(".", from_flax=True)
|
3 |
+
model.save_pretrained(".")
|
4 |
+
print("\nModel is now converted to PyTorch. There should be a new pytorch_model.bin-file in this directory.")
|
5 |
+
|
generation_config.json
CHANGED
@@ -220,5 +220,5 @@
|
|
220 |
"transcribe": 50359,
|
221 |
"translate": 50358
|
222 |
},
|
223 |
-
"transformers_version": "4.
|
224 |
}
|
|
|
220 |
"transcribe": 50359,
|
221 |
"translate": 50358
|
222 |
},
|
223 |
+
"transformers_version": "4.30.1"
|
224 |
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:d556f5a1c038eee4ec61beb097a4b668a65524ba2772778cc0b80d123d9ad32f
|
3 |
+
size 3055741593
|