(device, wav2vec)
| 254 | |
| 255 | |
| 256 | def custom_init(device, wav2vec): |
| 257 | audio_encoder = Wav2Vec2Model.from_pretrained(wav2vec, local_files_only=True).to(device) |
| 258 | audio_encoder.feature_extractor._freeze_parameters() |
| 259 | wav2vec_feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(wav2vec, local_files_only=True) |
| 260 | return wav2vec_feature_extractor, audio_encoder |
| 261 | |
| 262 | def loudness_norm(audio_array, sr=16000, lufs=-23): |
| 263 | meter = pyln.Meter(sr) |