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