30 lines
862 B
Python
30 lines
862 B
Python
|
from kotoba_whisper import KotobaWhisperPipeline
|
||
|
|
from transformers.pipelines import PIPELINE_REGISTRY, pipeline
|
||
|
|
from transformers import WhisperForConditionalGeneration, TFWhisperForConditionalGeneration
|
||
|
|
|
||
|
|
|
||
|
|
model_alias = "kotoba-tech/kotoba-whisper-v2.1"
|
||
|
|
PIPELINE_REGISTRY.register_pipeline(
|
||
|
|
"kotoba-whisper",
|
||
|
|
pipeline_class=KotobaWhisperPipeline,
|
||
|
|
pt_model=WhisperForConditionalGeneration,
|
||
|
|
tf_model=TFWhisperForConditionalGeneration
|
||
|
|
)
|
||
|
|
pipe = pipeline(
|
||
|
|
task="kotoba-whisper",
|
||
|
|
model="kotoba-tech/kotoba-whisper-v2.0",
|
||
|
|
chunk_length_s=15,
|
||
|
|
batch_size=16,
|
||
|
|
punctuator=True,
|
||
|
|
stable_ts=True,
|
||
|
|
)
|
||
|
|
pipe.push_to_hub(model_alias)
|
||
|
|
pipe = pipeline(model=model_alias,
|
||
|
|
punctuator=True,
|
||
|
|
stable_ts=True,
|
||
|
|
chunk_length_s=15,
|
||
|
|
batch_size=16,
|
||
|
|
trust_remote_code=True)
|
||
|
|
|
||
|
|
|