Automatic Speech Recognition
Transformers
Safetensors
Japanese
whisper
audio
hf-asr-leaderboard
asahi417 commited on
Commit
4d213e6
·
verified ·
1 Parent(s): c959a4f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -130,7 +130,7 @@ model_id = "kotoba-tech/kotoba-whisper-v1.1"
130
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
131
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
132
  model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
133
- generate_kwargs = {"language": "japanese", "task": "transcribe"}
134
 
135
  # load model
136
  pipe = pipeline(
 
130
  torch_dtype = torch.float16 if torch.cuda.is_available() else torch.float32
131
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
132
  model_kwargs = {"attn_implementation": "sdpa"} if torch.cuda.is_available() else {}
133
+ generate_kwargs = {"language": "ja", "task": "transcribe"}
134
 
135
  # load model
136
  pipe = pipeline(