kotoba-tech
/

kotoba-whisper-v2.0

Automatic Speech Recognition

hf-asr-leaderboard

Model card Files Files and versions

asahi417 commited on Sep 20, 2024

Commit

16d4df0

·

verified ·

1 Parent(s): f534806

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -142,7 +142,7 @@ model_id = "kotoba-tech/kotoba-whisper-v2.0"
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
-generate_kwargs = {"language": "japanese", "task": "transcribe"}
 # load model
 pipe = pipeline(
@@ -204,7 +204,7 @@ model_id = "kotoba-tech/kotoba-whisper-v2.0"
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
-generate_kwargs = {"language": "japanese", "task": "transcribe"}
 # load model
 pipe = pipeline(

 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
+generate_kwargs = {"language": "ja", "task": "transcribe"}
 # load model
 pipe = pipeline(
 torch_dtype = torch.bfloat16 if torch.cuda.is_available() else torch.float32
 device = "cuda:0" if torch.cuda.is_available() else "cpu"
 model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
+generate_kwargs = {"language": "ja", "task": "transcribe"}
 # load model
 pipe = pipeline(