detect_language() doesn't work on a GPU?

[utility-aagrawal]

detect_language() doesn't work on a GPU?

[utility-aagrawal]

What I don't understand is how transcribe() works even if I keep everything else in the code unchanged.

This code works:

import whisperx
import time

start_time = time.time()
filepath = ""

whisper_model = whisperx.load_model("medium", device = "cuda", compute_type="float16")

audio = whisperx.load_audio(filepath)
audio = whisperx.audio.pad_or_trim(audio)

results = whisper_model.transcribe(audio)
print(results)

end_time = time.time()
print(f"Time taken: {end_time - start_time:.2f} seconds")

transcribe() also uses the same detect_language() method as you can see here:

whisperX/whisperx/asr.py

Line 194 in f2da2f8

language = language or self.detect_language(audio)

but this doesn't throw the same error.