kotoba-tech
/

kotoba-whisper-v1.0

Automatic Speech Recognition

hf-asr-leaderboard

Inference Endpoints

Model card Files Files and versions Community

asahi417 commited on Sep 20, 2024

Commit

a81f3fe

·

verified ·

1 Parent(s): cb8ed34

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -141,7 +141,7 @@ model_id = "kotoba-tech/kotoba-whisper-v1.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(
@@ -239,7 +239,7 @@ model_id = "kotoba-tech/kotoba-whisper-v1.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(