kotoba-tech
/

kotoba-whisper-v1.1

Automatic Speech Recognition

hf-asr-leaderboard

Model card Files Files and versions

asahi417 commited on Sep 20, 2024

Commit

4d213e6

·

verified ·

1 Parent(s): c959a4f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -130,7 +130,7 @@ model_id = "kotoba-tech/kotoba-whisper-v1.1"
 torch_dtype = torch.float16 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.float16 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(