whisper-vs-distil-whisper-zero

Runtime error

sanchit-gandhi commited on Nov 2, 2023

Commit

26faa5f

1 Parent(s): 80ca55c

fix processor

Files changed (1) hide show

app.py CHANGED Viewed

@@ -25,7 +25,7 @@ if not use_flash_attention_2:
     model = model.to_bettertransformer()
     distilled_model = distilled_model.to_bettertransformer()
-processor = AutoProcessor.from_pretrained("openai/whisper-tiny.en")
 model.to(device)
 distilled_model.to(device)

     model = model.to_bettertransformer()
     distilled_model = distilled_model.to_bettertransformer()
+processor = AutoProcessor.from_pretrained("openai/whisper-large-v2")
 model.to(device)
 distilled_model.to(device)