avemio-digital commited on
Commit
7a18a1b
·
verified ·
1 Parent(s): 9c97ccc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -33,7 +33,7 @@ should probably proofread and complete it, then remove this comment. -->
33
 
34
  # GRAG-WHISPER-LARGE-v3-TURBO
35
 
36
- This model is a fine-tuned version of [openai/whisper-large-v3-turbo](https://huggingface.co/openai/whisper-large-v3-turbo) on a carefully curated dataset.
37
 
38
 
39
  ## Evaluations - Word error rate
@@ -58,7 +58,7 @@ from transformers import AutoModelForSpeechSeq2Seq, AutoProcessor, pipeline
58
  from datasets import load_dataset
59
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
60
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
61
- model_id = "primeline/whisper-large-v3-turbo-german"
62
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
63
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
64
  )
 
33
 
34
  # GRAG-WHISPER-LARGE-v3-TURBO
35
 
36
+ This model is a fine-tuned version of [openai/whisper-large-v3-turbo](https://huggingface.co/openai/whisper-large-v3-turbo) on a carefully curated 13 hour dataset.
37
 
38
 
39
  ## Evaluations - Word error rate
 
58
  from datasets import load_dataset
59
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
60
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
61
+ model_id = "avemio/GRAG-WHISPER-LARGE-v3-TURBO"
62
  model = AutoModelForSpeechSeq2Seq.from_pretrained(
63
  model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
64
  )