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
|