Update README.md
#2
by
muhtasham
- opened
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")
|