| """ | |
| Create the model configuration (`config.json`) and the feature extractor configuration | |
| (`preprocessor_config.json`). | |
| """ | |
| from transformers import Wav2Vec2Config, Wav2Vec2FeatureExtractor | |
| model_dir="./" | |
| config = Wav2Vec2Config.from_pretrained( | |
| "facebook/wav2vec2-base", | |
| mask_time_length=10, | |
| mask_time_prob=0.05, | |
| diversity_loss_weight=0.1, | |
| num_negatives=100, | |
| do_stable_layer_norm=True, | |
| feat_extract_norm="layer", | |
| ) | |
| config.save_pretrained(model_dir) | |
| config = Wav2Vec2FeatureExtractor.from_pretrained("facebook/wav2vec2-base", return_attention_mask=True) | |
| config.save_pretrained(model_dir) | |