Update extract_feature_print.py
Browse files- extract_feature_print.py +1 -1
extract_feature_print.py
CHANGED
|
@@ -122,7 +122,7 @@ model_configs = {
|
|
| 122 |
},
|
| 123 |
"wav2vec": {
|
| 124 |
"target_sr": 16000,
|
| 125 |
-
"load_model": lambda path, dev, half: load_huggingface_model(path, dev, half,
|
| 126 |
"preprocess": general_preprocess,
|
| 127 |
"prepare_input": general_prepare_input,
|
| 128 |
"extract_features": general_extract_features,
|
|
|
|
| 122 |
},
|
| 123 |
"wav2vec": {
|
| 124 |
"target_sr": 16000,
|
| 125 |
+
"load_model": lambda path, dev, half: load_huggingface_model(path, dev, half, Wav2Vec2Model),
|
| 126 |
"preprocess": general_preprocess,
|
| 127 |
"prepare_input": general_prepare_input,
|
| 128 |
"extract_features": general_extract_features,
|