(device, wav2vec)
| 275 | return args |
| 276 | |
| 277 | def custom_init(device, wav2vec): |
| 278 | audio_encoder = Wav2Vec2Model.from_pretrained(wav2vec, local_files_only=True).to(device) |
| 279 | audio_encoder.feature_extractor._freeze_parameters() |
| 280 | wav2vec_feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(wav2vec, local_files_only=True) |
| 281 | return wav2vec_feature_extractor, audio_encoder |
| 282 | |
| 283 | def loudness_norm(audio_array, sr=16000, lufs=-23): |
| 284 | meter = pyln.Meter(sr) |