TechInterMezzo
/

whisper-encoder-medium

Model card Files Files and versions

TechInterMezzo commited on Nov 5, 2023

Commit

1609f1f

·

1 Parent(s): 37f94c5

Update README.md

Files changed (1) hide show

README.md +3 -2

README.md CHANGED Viewed

@@ -10,6 +10,7 @@ feature_extractor = WhisperFeatureExtractor.from_pretrained("techintermezzo/whis
 model = WhisperEncoder.from_pretrained("techintermezzo/whisper-encoder-medium").half()
 model.eval()
-input_features = feature_extractor(inputs, sampling_rate=16000, return_tensors="pt").input_features
-last_hidden_state = model(input_features).last_hidden_state
 ```

 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
 ```