Generated inference code uses AutoModelForSeq2SeqLM, which breaks for Whisper

by irudachirath - opened Jan 11

SPEAK-ASR org Jan 11

Problem

The auto-generated inference notebook/code on the model page is incorrect.

It uses:

from transformers import AutoModelForSeq2SeqLM

base_model = AutoModelForSeq2SeqLM.from_pretrained("openai/whisper-small")

Upload images, audio, and videos by dragging in the text input, pasting, or clicking here.

Tap or paste here to upload images

· Sign up or log in to comment