openai/whisper-large-v3-turbo
openai/whisper-large-v3-turbo
2
1
Overview Code Discussions Pull Requests Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Yoach Lacombe 2024-10-01 08:38:45 +00:00 committed by system
parent d427ac9d61
commit fd03fade0e
No known key found for this signature in database
GPG Key ID: 6A528E38E0733467
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
README.md
View File

@ -148,7 +148,7 @@ from datasets import load_dataset
device = "cuda:0" if torch.cuda.is_available() else "cpu" device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
model_id = "openai/whisper-large-v3-turbo" model_id = "ylacombe/whisper-large-v3-turbo"
model = AutoModelForSpeechSeq2Seq.from_pretrained( model = AutoModelForSpeechSeq2Seq.from_pretrained(
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True, use_safetensors=True
@ -252,7 +252,7 @@ from datasets import Audio, load_dataset
device = "cuda:0" if torch.cuda.is_available() else "cpu" device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
model_id = "openai/whisper-large-v3-turbo" model_id = "ylacombe/whisper-large-v3-turbo"
model = AutoModelForSpeechSeq2Seq.from_pretrained( model = AutoModelForSpeechSeq2Seq.from_pretrained(
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True
@ -327,7 +327,7 @@ from datasets import load_dataset
device = "cuda:0" if torch.cuda.is_available() else "cpu" device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
model_id = "openai/whisper-large-v3-turbo" model_id = "ylacombe/whisper-large-v3-turbo"
model = AutoModelForSpeechSeq2Seq.from_pretrained( model = AutoModelForSpeechSeq2Seq.from_pretrained(
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True
@ -373,7 +373,7 @@ torch.set_float32_matmul_precision("high")
device = "cuda:0" if torch.cuda.is_available() else "cpu" device = "cuda:0" if torch.cuda.is_available() else "cpu"
torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32 torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
model_id = "openai/whisper-large-v3-turbo" model_id = "ylacombe/whisper-large-v3-turbo"
model = AutoModelForSpeechSeq2Seq.from_pretrained( model = AutoModelForSpeechSeq2Seq.from_pretrained(
model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True model_id, torch_dtype=torch_dtype, low_cpu_mem_usage=True
@ -472,7 +472,7 @@ checkpoints are summarised in the following table with links to the models on th
| large | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large) | | large | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large) |
| large-v2 | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large-v2) | | large-v2 | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large-v2) |
| large-v3 | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large-v3) | | large-v3 | 1550 M | x | [✓](https://huggingface.co/openai/whisper-large-v3) |
| large-v3-turbo | 809 M | x | [✓](https://huggingface.co/openai/whisper-large-v3-turbo) | | large-v3-turbo | 809 M | x | [✓](https://huggingface.co/ylacombe/whisper-large-v3-turbo) |
## Fine-Tuning ## Fine-Tuning