mirror of
https://github.com/openai/whisper.git
synced 2025-11-24 14:35:57 +00:00
Merge 66f30a3d3c41a0ffe9f84554b415c22a45b807a7 into c0d2f624c09dc18e709e37c2ad90c039a4eb72a2
This commit is contained in:
commit
5ab934e831
@ -514,7 +514,7 @@ class DecodingTask:
|
|||||||
def __init__(self, model: "Whisper", options: DecodingOptions):
|
def __init__(self, model: "Whisper", options: DecodingOptions):
|
||||||
self.model = model
|
self.model = model
|
||||||
|
|
||||||
language = options.language or "en"
|
language = options.language or "default"
|
||||||
tokenizer = get_tokenizer(
|
tokenizer = get_tokenizer(
|
||||||
model.is_multilingual,
|
model.is_multilingual,
|
||||||
num_languages=model.num_languages,
|
num_languages=model.num_languages,
|
||||||
|
|||||||
@ -381,7 +381,7 @@ def get_tokenizer(
|
|||||||
|
|
||||||
if multilingual:
|
if multilingual:
|
||||||
encoding_name = "multilingual"
|
encoding_name = "multilingual"
|
||||||
language = language or "en"
|
language = language or "default"
|
||||||
task = task or "transcribe"
|
task = task or "transcribe"
|
||||||
else:
|
else:
|
||||||
encoding_name = "gpt2"
|
encoding_name = "gpt2"
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user