(device, wav2vec)
| 244 | |
| 245 | |
| 246 | def custom_init(device, wav2vec): |
| 247 | audio_encoder = Wav2Vec2Model.from_pretrained(wav2vec, local_files_only=True).to(device) |
| 248 | audio_encoder.feature_extractor._freeze_parameters() |
| 249 | wav2vec_feature_extractor = Wav2Vec2FeatureExtractor.from_pretrained(wav2vec, local_files_only=True) |
| 250 | return wav2vec_feature_extractor, audio_encoder |
| 251 | |
| 252 | def loudness_norm(audio_array, sr=16000, lufs=-23): |
| 253 | meter = pyln.Meter(sr) |