Views
No views yet
1from transformers import WhisperFeatureExtractor
2from transformers.models.whisper.modeling_whisper import WhisperEncoder
3
4feature_extractor = WhisperFeatureExtractor.from_pretrained("techintermezzo/whisper-encoder-medium")
5model = WhisperEncoder.from_pretrained("techintermezzo/whisper-encoder-medium").half()
6model.eval()
7
8with torch.inference_mode():
9 input_features = feature_extractor(inputs, sampling_rate=16000, return_tensors="pt").input_features
10 last_hidden_state = model(input_features).last_hidden_state