Automatic Speech Recognition
Transformers
Safetensors
Japanese
whisper
audio
hf-asr-leaderboard
asahi417 commited on
Commit
3151dd8
·
verified ·
1 Parent(s): 2589d37

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -108,7 +108,7 @@ model_id = "kotoba-tech/kotoba-whisper-v2.1"
108
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
109
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
110
  model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
111
- generate_kwargs = {"language": "japanese", "task": "transcribe"}
112
 
113
  # load model
114
  pipe = pipeline(
 
108
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
109
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
110
  model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
111
+ generate_kwargs = {"language": "ja", "task": "transcribe"}
112
 
113
  # load model
114
  pipe = pipeline(