Hugging Face's logo

jordand
/
whisper-d-v1a

PyTorch
whisper
Model card Files Community
1
jordand commited on
Commit
80e3308
·
verified ·
1 Parent(s): 822fb00

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -2
README.md CHANGED
@@ -26,8 +26,7 @@ import torch
26
  import torchaudio
27
  from transformers import WhisperForConditionalGeneration, WhisperProcessor, WhisperTokenizer
28
 
29
- model = WhisperForConditionalGeneration.from_pretrained('jordand/whisper-d-v1a',
30
- torch_dtype=torch.float16).cuda()
31
  processor = WhisperProcessor.from_pretrained('openai/whisper-large-v2')
32
  tokenizer = WhisperTokenizer.from_pretrained('openai/whisper-large-v2')
33
 
 
26
  import torchaudio
27
  from transformers import WhisperForConditionalGeneration, WhisperProcessor, WhisperTokenizer
28
 
29
+ model = WhisperForConditionalGeneration.from_pretrained('jordand/whisper-d-v1a', torch_dtype=torch.float16).cuda()
 
30
  processor = WhisperProcessor.from_pretrained('openai/whisper-large-v2')
31
  tokenizer = WhisperTokenizer.from_pretrained('openai/whisper-large-v2')
32