File size: 254 Bytes
f1a1d63
 
 
 
 
1
2
3
4
5
# Load model directly
from transformers import AutoProcessor, AutoModelForSpeechSeq2Seq

processor = AutoProcessor.from_pretrained("pedrogarcias/whisper-small-v2-jax")
model = AutoModelForSpeechSeq2Seq.from_pretrained("pedrogarcias/whisper-small-v2-jax")