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