masoudkaviani commited on
Commit
ba33076
·
verified ·
1 Parent(s): 158e67d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -49,7 +49,7 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
49
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
50
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
51
 
52
- model_id = "aictsharif/whisper-tiny-fa"
53
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
54
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
55
  )
 
49
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
50
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
51
 
52
+ model_id = "aictsharif/whisper-large-v2-fa"
53
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
54
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
55
  )