|
--- |
|
license: mit |
|
--- |
|
|
|
```python |
|
from transformers import WhisperFeatureExtractor |
|
from transformers.models.whisper.modeling_whisper import WhisperEncoder |
|
|
|
feature_extractor = WhisperFeatureExtractor.from_pretrained("techintermezzo/whisper-encoder-medium") |
|
model = WhisperEncoder.from_pretrained("techintermezzo/whisper-encoder-medium").half() |
|
model.eval() |
|
|
|
with torch.inference_mode(): |
|
input_features = feature_extractor(inputs, sampling_rate=16000, return_tensors="pt").input_features |
|
last_hidden_state = model(input_features).last_hidden_state |
|
``` |