Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -42,7 +42,7 @@ device = "cuda:0" if torch.cuda.is_available() else "cpu"
42
  compute_type = "float16" if torch.cuda.is_available() else "float32"
43
 
44
  # load model on GPU if available, else cpu
45
- model = WhisperModel("distil-large-v3", device=device, compute_type=compute_type)
46
 
47
  # load toy dataset for example
48
  dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")
 
42
  compute_type = "float16" if torch.cuda.is_available() else "float32"
43
 
44
  # load model on GPU if available, else cpu
45
+ model = WhisperModel("distil-whisper/distil-large-v3-ct2", device=device, compute_type=compute_type)
46
 
47
  # load toy dataset for example
48
  dataset = load_dataset("hf-internal-testing/librispeech_asr_dummy", "clean", split="validation")