Sharfraz's picture
Create v2 jax
f1a1d63 verified
raw
history blame contribute delete
254 Bytes
# 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")