Automatic Speech Recognition
Transformers
PyTorch
JAX
TensorBoard
Safetensors
Norwegian
whisper
audio
asr
hf-asr-leaderboard
Instructions to use NbAiLabArchive/scream_medium_beta with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use NbAiLabArchive/scream_medium_beta with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("automatic-speech-recognition", model="NbAiLabArchive/scream_medium_beta")# Load model directly from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq processor = AutoProcessor.from_pretrained("NbAiLabArchive/scream_medium_beta") model = AutoModelForSpeechSeq2Seq.from_pretrained("NbAiLabArchive/scream_medium_beta") - Notebooks
- Google Colab
- Kaggle
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
|