Change faster-whisper as default

This commit is contained in:
litagin02
2024-11-11 10:25:30 +09:00
parent 038d07f4e7
commit 875186f0c7

View File

@@ -174,12 +174,13 @@ def create_dataset_app() -> gr.Blocks:
)
use_hf_whisper = gr.Checkbox(
label="HuggingFaceのWhisperを使う速度が速いがVRAMを多く使う",
value=True,
value=False,
)
hf_repo_id = gr.Dropdown(
["openai/whisper", "kotoba-tech/kotoba-whisper-v1.1"],
label="HuggingFaceのWhisperモデル",
value="openai/whisper",
visible=False,
)
compute_type = gr.Dropdown(
[
@@ -194,7 +195,7 @@ def create_dataset_app() -> gr.Blocks:
],
label="計算精度",
value="bfloat16",
visible=False,
visible=True,
)
batch_size = gr.Slider(
minimum=1,
@@ -203,6 +204,7 @@ def create_dataset_app() -> gr.Blocks:
step=1,
label="バッチサイズ",
info="大きくすると速度が速くなるがVRAMを多く使う",
visible=False,
)
language = gr.Dropdown(["ja", "en", "zh"], value="ja", label="言語")
initial_prompt = gr.Textbox(