PyTorch model 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,
|
@@ -144,7 +144,7 @@
|
|
144 |
50362
|
145 |
],
|
146 |
"torch_dtype": "float32",
|
147 |
-
"transformers_version": "4.
|
148 |
"use_cache": true,
|
149 |
"use_weighted_layer_sum": false,
|
150 |
"vocab_size": 51865
|
|
|
1 |
{
|
2 |
+
"_name_or_path": ".",
|
3 |
"activation_dropout": 0.0,
|
4 |
"activation_function": "gelu",
|
5 |
"apply_spec_augment": false,
|
|
|
144 |
50362
|
145 |
],
|
146 |
"torch_dtype": "float32",
|
147 |
+
"transformers_version": "4.30.1",
|
148 |
"use_cache": true,
|
149 |
"use_weighted_layer_sum": false,
|
150 |
"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
@@ -221,5 +221,5 @@
|
|
221 |
"transcribe": 50359,
|
222 |
"translate": 50358
|
223 |
},
|
224 |
-
"transformers_version": "4.
|
225 |
}
|
|
|
221 |
"transcribe": 50359,
|
222 |
"translate": 50358
|
223 |
},
|
224 |
+
"transformers_version": "4.30.1"
|
225 |
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1d245f6040b490abff283959ce7dbddb5a7d0a485814ce749f5e0a305bd95c78
|
3 |
+
size 967095881
|