Update README.md
Browse files
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": "
|
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(
|